Spaces:
Running
on
Zero
Running
on
Zero
use smaller model
Browse files
app.py
CHANGED
@@ -20,12 +20,12 @@ os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
|
|
20 |
|
21 |
|
22 |
model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
23 |
-
"Qwen/Qwen2.5-VL-
|
24 |
torch_dtype=torch.bfloat16,
|
25 |
attn_implementation="flash_attention_2",
|
26 |
device_map="auto",
|
27 |
)
|
28 |
-
processor = AutoProcessor.from_pretrained("Qwen/Qwen2.5-VL-
|
29 |
|
30 |
|
31 |
class GeneralRetrievalQuery(BaseModel):
|
|
|
20 |
|
21 |
|
22 |
model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
23 |
+
"Qwen/Qwen2.5-VL-3B-Instruct",
|
24 |
torch_dtype=torch.bfloat16,
|
25 |
attn_implementation="flash_attention_2",
|
26 |
device_map="auto",
|
27 |
)
|
28 |
+
processor = AutoProcessor.from_pretrained("Qwen/Qwen2.5-VL-3B-Instruct")
|
29 |
|
30 |
|
31 |
class GeneralRetrievalQuery(BaseModel):
|