diff --git a/tools/inference_engine/reference_loader.py b/tools/inference_engine/reference_loader.py index 6369f1db..6f19c6dd 100644 --- a/tools/inference_engine/reference_loader.py +++ b/tools/inference_engine/reference_loader.py @@ -94,9 +94,7 @@ def load_by_hash( 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: