Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -100,12 +100,12 @@ def load_image(image_file, input_size=448, max_num=12):
|
|
| 100 |
return pixel_values
|
| 101 |
|
| 102 |
model = AutoModel.from_pretrained(
|
| 103 |
-
"5CD-AI/Vintern-
|
| 104 |
torch_dtype=torch.bfloat16,
|
| 105 |
low_cpu_mem_usage=True,
|
| 106 |
trust_remote_code=True,
|
| 107 |
).eval().cuda()
|
| 108 |
-
tokenizer = AutoTokenizer.from_pretrained("5CD-AI/Vintern-
|
| 109 |
|
| 110 |
|
| 111 |
@spaces.GPU
|
|
|
|
| 100 |
return pixel_values
|
| 101 |
|
| 102 |
model = AutoModel.from_pretrained(
|
| 103 |
+
"5CD-AI/Vintern-1B-v3_5",
|
| 104 |
torch_dtype=torch.bfloat16,
|
| 105 |
low_cpu_mem_usage=True,
|
| 106 |
trust_remote_code=True,
|
| 107 |
).eval().cuda()
|
| 108 |
+
tokenizer = AutoTokenizer.from_pretrained("5CD-AI/Vintern-1B-v3_5", trust_remote_code=True, use_fast=False)
|
| 109 |
|
| 110 |
|
| 111 |
@spaces.GPU
|