Update app.py
Browse files
app.py
CHANGED
|
@@ -325,7 +325,7 @@ def init_dinov2(model_name: str = DINOV2_MODEL, device: torch.device = DEVICE):
|
|
| 325 |
global _dinov2_processor, _dinov2_model
|
| 326 |
if _dinov2_processor is None or _dinov2_model is None:
|
| 327 |
# _dinov2_processor = AutoImageProcessor.from_pretrained(model_name)
|
| 328 |
-
_dinov2_processor = AutoImageProcessor.from_pretrained(
|
| 329 |
_dinov2_model = AutoModel.from_pretrained(model_name)
|
| 330 |
_dinov2_model.eval().to(device)
|
| 331 |
|
|
|
|
| 325 |
global _dinov2_processor, _dinov2_model
|
| 326 |
if _dinov2_processor is None or _dinov2_model is None:
|
| 327 |
# _dinov2_processor = AutoImageProcessor.from_pretrained(model_name)
|
| 328 |
+
_dinov2_processor = AutoImageProcessor.from_pretrained(model_name, use_fast=True)
|
| 329 |
_dinov2_model = AutoModel.from_pretrained(model_name)
|
| 330 |
_dinov2_model.eval().to(device)
|
| 331 |
|