Update minicpm-o_audio.py

7326adc9 · wanglch · 79b242e1 · 7326adc9
Commit 7326adc9 authored Apr 15, 2025 by wanglch
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

minicpm-o_audio.py minicpm-o_audio.py +1 -1

No files found.
--- a/minicpm-o_audio.py
+++ b/minicpm-o_audio.py
@@ -15,7 +15,7 @@ from modelscope import AutoModel, AutoTokenizer
 model = AutoModel.from_pretrained(
    'openbmb/MiniCPM-o-2_6',
    trust_remote_code=True,
-    attn_implementation='sdpa', # sdpa or flash_attention_2
+    attn_implementation='eager', # sdpa or flash_attention_2
    torch_dtype=torch.bfloat16,
    init_vision=True,
    init_audio=True,