Spaces:

YueMafighting
/

FollowYourPose

Runtime error

YueMafighting commited on Apr 7, 2023

Commit

cf9f29a

•

1 Parent(s): 4750478

Update FollowYourPose/followyourpose/pipelines/pipeline_followyourpose.py

Files changed (1) hide show

FollowYourPose/followyourpose/pipelines/pipeline_followyourpose.py CHANGED Viewed

@@ -419,7 +419,7 @@ class FollowYourPosePipeline(DiffusionPipeline):
     @torch.no_grad()
     def get_skeleton(self,skeleton_path):
-        skeleton_start_end = list(range(0, 60, 5))
         self_transform = transforms.Compose([transforms.Resize(512),
                                         transforms_video.CenterCropVideo(512)])

     @torch.no_grad()
     def get_skeleton(self,skeleton_path):
+        skeleton_start_end = list(range(0, 16, 2))
         self_transform = transforms.Compose([transforms.Resize(512),
                                         transforms_video.CenterCropVideo(512)])