Procházet zdrojové kódy

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

Hao Guan před 1 rokem
rodič
revize
fd0b73e699
1 změnil soubory, kde provedl 1 přidání a 3 odebrání
  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: