Spaces:
Running
on
Zero
Running
on
Zero
khang119966
commited on
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
|