RamAnanth1 commited on
Commit
03bc7e7
1 Parent(s): c6e2b5c

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -89,7 +89,7 @@ class Model:
89
  self.sampler= PLMSSampler(self.model)
90
  save_memory=True
91
 
92
- self.model_ad_pose = Adapter(channels=[320, 640, 1280, 1280][:4], nums_rb=2, ksize=1, sk=True, use_conv=False).to(device)
93
  self.model_ad_pose.load_state_dict(torch.load("models/t2iadapter_keypose_sd14v1.pth"))
94
 
95
 
 
89
  self.sampler= PLMSSampler(self.model)
90
  save_memory=True
91
 
92
+ self.model_ad_pose = Adapter(cin=int(3*64),channels=[320, 640, 1280, 1280][:4], nums_rb=2, ksize=1, sk=True, use_conv=False).to(device)
93
  self.model_ad_pose.load_state_dict(torch.load("models/t2iadapter_keypose_sd14v1.pth"))
94
 
95