Spaces:
Running
Running
Update app.py
Browse files
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='
|
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()
|