Spaces:

Adapter
/

T2I-Adapter

Runtime error

Adapter commited on Feb 24, 2023

Commit

132e5ba

1 Parent(s): 7194abe

Update demo/model.py

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 = im_keypose_out.astype(np.uint8)
         # 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)[:,:,::-1]
         # extract condition features
         c = self.base_model.get_learned_conditioning([prompt + ', ' + pos_prompt])