Spaces:
Running
Running
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(
|