Hugging Face's logo

Spaces:
wondervictor
/
ControlAR
Running on Zero

App Files Community
1
wondervictor commited on
Commit
0203683
·
verified ·
1 Parent(s): 6b11bf0

Update autoregressive/models/dinov2_adapter.py

Browse files
Files changed (1) hide show
  1. autoregressive/models/dinov2_adapter.py +1 -1
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.from_pretrained(f'facebook/dinov2-{adapter_size}')
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):