소스 검색

fix: cache reference by hash (#730) (#741)

Hao Guan 1 년 전
부모
커밋
fd0b73e699
1개의 변경된 파일1개의 추가작업 그리고 3개의 파일을 삭제
  1. 1 3
      tools/inference_engine/reference_loader.py

+ 1 - 3
tools/inference_engine/reference_loader.py

@@ -94,9 +94,7 @@ class ReferenceLoader:
 
             else:
                 # Reuse already encoded references
-                prompt_text, prompt_token = self.ref_by_hash[audio_hashes[i]]
-                prompt_texts.append(prompt_text)
-                prompt_tokens.append(prompt_token)
+                prompt_tokens, prompt_texts = self.ref_by_hash[audio_hashes[i]]
                 cache_used = True
 
         if cache_used: