Update app.py
Browse files
app.py
CHANGED
|
@@ -324,7 +324,8 @@ def init_dinov2(model_name: str = DINOV2_MODEL, device: torch.device = DEVICE):
|
|
| 324 |
"""Lazy-initialize DINOv2 processor & model (call once before embedding)."""
|
| 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_model = AutoModel.from_pretrained(model_name)
|
| 329 |
_dinov2_model.eval().to(device)
|
| 330 |
|
|
|
|
| 324 |
"""Lazy-initialize DINOv2 processor & model (call once before embedding)."""
|
| 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_or_path, use_fast=True)
|
| 329 |
_dinov2_model = AutoModel.from_pretrained(model_name)
|
| 330 |
_dinov2_model.eval().to(device)
|
| 331 |
|