fix a bug

Signed-off-by: Gaohan123 <gaohan19@huawei.com>
This commit is contained in:
Gaohan123 committed 2025-11-05 13:05:12 +08:00
1 parent c80348ce71
commit e2b803fbe6
1 file changed
+2 -1
@@ -818,10 +818,11 @@ class Qwen2_5OmniForConditionalGeneration(
# Load token2wav weights (if any)
if token2wav_weights and self.token2wav is not None:
# download weights from huggingface for spk_dict.pt
hf_model_folder = download_weights_from_hf_specific(
self.vllm_config.model_config.model,
self.vllm_config.load_config.download_dir,
allow_patterns=["*.safetensors", "*.bin", "*.pt"],
allow_patterns=["*.pt"],
)
self._init_token2wav_model(hf_model_folder)
t2w_loaded = self.token2wav.load_weights(