Spaces:
Runtime error
Runtime error
Update main.py
Browse files
main.py
CHANGED
|
@@ -36,7 +36,7 @@ class PredictRequest(BaseModel):
|
|
| 36 |
# # attn_implementation="flash_attention_2",
|
| 37 |
# )
|
| 38 |
# checkpoint = "Qwen/Qwen2.5-VL-7B-Instruct"
|
| 39 |
-
checkpoint = "Qwen/Qwen2.5-VL-
|
| 40 |
min_pixels = 256 * 28 * 28
|
| 41 |
max_pixels = 1280 * 28 * 28
|
| 42 |
processor = AutoProcessor.from_pretrained(
|
|
|
|
| 36 |
# # attn_implementation="flash_attention_2",
|
| 37 |
# )
|
| 38 |
# checkpoint = "Qwen/Qwen2.5-VL-7B-Instruct"
|
| 39 |
+
checkpoint = "Qwen/Qwen2.5-VL-3B-Instruct-AWQ"
|
| 40 |
min_pixels = 256 * 28 * 28
|
| 41 |
max_pixels = 1280 * 28 * 28
|
| 42 |
processor = AutoProcessor.from_pretrained(
|