fix: not force reloading dinov2 models

jolibrain · Nov 24, 2023 · 800967d · 800967d
1 parent 634d48a
commit 800967d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/modules/projected_d/projector.py b/models/modules/projected_d/projector.py
@@ -274,7 +274,7 @@ def create_clip_model(model_name, config_path, weight_path, img_size):
 
 def create_dinov2_model(model_name, config_path, weight_path, img_size):
     dinov2_model = torch.hub.load(
-        "facebookresearch/dinov2", model_name, force_reload=True
+        "facebookresearch/dinov2", model_name, force_reload=False
     )
     return dinov2_model