Spaces:

Adapter
/

T2I-Adapter

Runtime error

Adapter commited on Feb 24, 2023

Commit

7194abe

1 Parent(s): 5d485d0

fix bug

Files changed (1) hide show

demo/model.py CHANGED Viewed

@@ -378,7 +378,7 @@ class Model_all:
                 pose_link_color=self.pose_link_color,
                 radius=2,
                 thickness=2)
-            im_keypose_out = cv2.resize(im_keypose_out, (512, 512))
         # extract condition features
         c = self.base_model.get_learned_conditioning([prompt + ', ' + pos_prompt])

                 pose_link_color=self.pose_link_color,
                 radius=2,
                 thickness=2)
+            im_keypose_out = im_keypose_out.astype(np.uint8)
         # extract condition features
         c = self.base_model.get_learned_conditioning([prompt + ', ' + pos_prompt])