Update autoregressive/models/dinov2_adapter.py

autoregressive/models/dinov2_adapter.py

@@ -10,7 +10,7 @@ class Dinov2_Adapter(nn.Module):
         super(Dinov2_Adapter, self).__init__()
         print(f"Choose adapter size: {adapter_size}")
         print(f"condition type: {condition_type}")
-        self.model = AutoModel.from_pretrained(f'facebook/dinov2-{adapter_size}')
+        self.model = AutoModel.from_pretrained(f'checkpoints/dinov2-{adapter_size}')
         # config = AutoConfig.from_pretrained(f'facebook/dinov2-{adapter_size}')
         # self.model = AutoModel.from_config(config)
         self.condition_type = condition_type