Spaces:
Running
on
Zero
Running
on
Zero
Antoine Chaffin
commited on
Commit
·
5885f58
1
Parent(s):
30c1e0f
Changing from 2B to 7B
Browse files
app.py
CHANGED
|
@@ -20,14 +20,14 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
| 20 |
# Initialize the model and processor
|
| 21 |
model = (
|
| 22 |
Qwen2VLForConditionalGeneration.from_pretrained(
|
| 23 |
-
"Qwen/Qwen2-VL-
|
| 24 |
)
|
| 25 |
.to(device)
|
| 26 |
.eval()
|
| 27 |
)
|
| 28 |
|
| 29 |
processor = AutoProcessor.from_pretrained(
|
| 30 |
-
"Qwen/Qwen2-VL-
|
| 31 |
)
|
| 32 |
|
| 33 |
|
|
|
|
| 20 |
# Initialize the model and processor
|
| 21 |
model = (
|
| 22 |
Qwen2VLForConditionalGeneration.from_pretrained(
|
| 23 |
+
"Qwen/Qwen2-VL-7B-Instruct", trust_remote_code=True, torch_dtype=torch.bfloat16
|
| 24 |
)
|
| 25 |
.to(device)
|
| 26 |
.eval()
|
| 27 |
)
|
| 28 |
|
| 29 |
processor = AutoProcessor.from_pretrained(
|
| 30 |
+
"Qwen/Qwen2-VL-7B-Instruct", trust_remote_code=True
|
| 31 |
)
|
| 32 |
|
| 33 |
|