Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -153,7 +153,7 @@ class ModelManager:
|
|
| 153 |
self.vision_model = AutoModel.from_pretrained(
|
| 154 |
base_vision_model_name, torch_dtype=torch.float16 if DEVICE == "cuda" else torch.float32
|
| 155 |
).to(DEVICE).eval()
|
| 156 |
-
else:
|
| 157 |
self.hf_processor = AutoProcessor.from_pretrained(base_vision_model_name)
|
| 158 |
self.vision_model = AutoModel.from_pretrained(
|
| 159 |
base_vision_model_name,
|
|
|
|
| 153 |
self.vision_model = AutoModel.from_pretrained(
|
| 154 |
base_vision_model_name, torch_dtype=torch.float16 if DEVICE == "cuda" else torch.float32
|
| 155 |
).to(DEVICE).eval()
|
| 156 |
+
else:
|
| 157 |
self.hf_processor = AutoProcessor.from_pretrained(base_vision_model_name)
|
| 158 |
self.vision_model = AutoModel.from_pretrained(
|
| 159 |
base_vision_model_name,
|