Spaces:
Runtime error
Runtime error
Update autoregressive/models/dinov2_adapter.py
Browse files
autoregressive/models/dinov2_adapter.py
CHANGED
|
@@ -10,7 +10,7 @@ class Dinov2_Adapter(nn.Module):
|
|
| 10 |
super(Dinov2_Adapter, self).__init__()
|
| 11 |
print(f"Choose adapter size: {adapter_size}")
|
| 12 |
print(f"condition type: {condition_type}")
|
| 13 |
-
self.model = AutoModel
|
| 14 |
self.condition_type = condition_type
|
| 15 |
|
| 16 |
def to_patch14(self, input):
|
|
|
|
| 10 |
super(Dinov2_Adapter, self).__init__()
|
| 11 |
print(f"Choose adapter size: {adapter_size}")
|
| 12 |
print(f"condition type: {condition_type}")
|
| 13 |
+
self.model = AutoModel()#.from_pretrained(f'facebook/dinov2-{adapter_size}')
|
| 14 |
self.condition_type = condition_type
|
| 15 |
|
| 16 |
def to_patch14(self, input):
|