Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -130,7 +130,7 @@ model_v = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
|
| 130 |
MODEL_ID_V,
|
| 131 |
trust_remote_code=True,
|
| 132 |
torch_dtype=torch.float16,
|
| 133 |
-
_attn_implementation="flash_attention_2"
|
| 134 |
).to(device).eval()
|
| 135 |
|
| 136 |
# Load Dots.OCR
|
|
|
|
| 130 |
MODEL_ID_V,
|
| 131 |
trust_remote_code=True,
|
| 132 |
torch_dtype=torch.float16,
|
| 133 |
+
#_attn_implementation="flash_attention_2"
|
| 134 |
).to(device).eval()
|
| 135 |
|
| 136 |
# Load Dots.OCR
|