Add trust_remote_code=True for Qwen2.5-Omni-7B loading

parent 994dbc48
Pipeline #204 canceled with stages
......@@ -74,8 +74,8 @@ def main():
return
# Load Qwen2.5-Omni-7B model
processor = AutoProcessor.from_pretrained("Qwen/Qwen2.5-Omni-7B")
model = AutoModel.from_pretrained("Qwen/Qwen2.5-Omni-7B")
processor = AutoProcessor.from_pretrained("Qwen/Qwen2.5-Omni-7B", trust_remote_code=True)
model = AutoModel.from_pretrained("Qwen/Qwen2.5-Omni-7B", trust_remote_code=True)
# Load audio
audio, sr = librosa.load(audio_file, sr=16000)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment