Spaces:
Runtime error
Runtime error
Commit
·
03bc7e7
1
Parent(s):
c6e2b5c
Update model.py
Browse files
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 |
|