Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -68,7 +68,7 @@ model = ColPali.from_pretrained(
|
|
| 68 |
"google/paligemma-3b-mix-448", torch_dtype=torch.bfloat16, device_map="cuda", token = token).eval()
|
| 69 |
|
| 70 |
model.load_adapter(model_name)
|
| 71 |
-
processor = AutoProcessor.from_pretrained(model_name,
|
| 72 |
|
| 73 |
device = model.device
|
| 74 |
|
|
|
|
| 68 |
"google/paligemma-3b-mix-448", torch_dtype=torch.bfloat16, device_map="cuda", token = token).eval()
|
| 69 |
|
| 70 |
model.load_adapter(model_name)
|
| 71 |
+
processor = AutoProcessor.from_pretrained(model_name, token = token)
|
| 72 |
|
| 73 |
device = model.device
|
| 74 |
|