Update autoregressive/models/dinov2_adapter.py

autoregressive/models/dinov2_adapter.py

@@ -1,4 +1,4 @@
-from transformers import AutoImageProcessor, AutoModel
+from transformers import AutoImageProcessor, AutoModel, AutoConfig
 from PIL import Image
 import requests
 import torch
@@ -10,7 +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}')
+        # 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):