Spaces:

RamAnanth1
/

T2I-Adapter

Runtime error

RamAnanth1 commited on Feb 21, 2023

Commit

d059ff8

•

1 Parent(s): c2ce6f5

bug fix

Files changed (1) hide show

model.py CHANGED Viewed

@@ -207,7 +207,7 @@ class Model:
         self.model_ad_pose.load_state_dict(torch.load("models/t2iadapter_keypose_sd14v1.pth"))
         self.model_ad_seg = Adapter(cin=int(3*64),channels=[320, 640, 1280, 1280][:4], nums_rb=2, ksize=1, sk=True, use_conv=False).to(device)
-        self.model_ad_seg.load_state_dict(torch.load("models/t2iadapter_seg_sd14v1.pth""))
     @torch.inference_mode()

         self.model_ad_pose.load_state_dict(torch.load("models/t2iadapter_keypose_sd14v1.pth"))
         self.model_ad_seg = Adapter(cin=int(3*64),channels=[320, 640, 1280, 1280][:4], nums_rb=2, ksize=1, sk=True, use_conv=False).to(device)
+        self.model_ad_seg.load_state_dict(torch.load("models/t2iadapter_seg_sd14v1.pth"))
     @torch.inference_mode()