gron371 commited on
Commit
b15e27e
·
verified ·
1 Parent(s): edd8504

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ async def init_model():
24
  model = AutoModel.from_pretrained(
25
  'openbmb/MiniCPM-V-2_6',
26
  trust_remote_code=True,
27
- attn_implementation='sdpa',
28
  torch_dtype=torch.bfloat16
29
  )
30
  model = model.eval()
 
24
  model = AutoModel.from_pretrained(
25
  'openbmb/MiniCPM-V-2_6',
26
  trust_remote_code=True,
27
+ attn_implementation='eager',
28
  torch_dtype=torch.bfloat16
29
  )
30
  model = model.eval()