Update autoregressive/models/dinov2_adapter.py

autoregressive/models/dinov2_adapter.py

@@ -10,9 +10,9 @@ 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}')
-        config = AutoConfig.from_pretrained(f'facebook/dinov2-{adapter_size}')
-        self.model = AutoModel.from_config(config)
+        self.model = AutoModel.from_pretrained(f'facebook/dinov2-{adapter_size}')
+        # config = AutoConfig.from_pretrained(f'facebook/dinov2-{adapter_size}')
+        # self.model = AutoModel.from_config(config)
         self.condition_type = condition_type
     
     def to_patch14(self, input):