first release

FollowYourPose/followyourpose/models/unet.py

@@ -211,7 +211,7 @@ class UNet3DConditionModel(ModelMixin, ConfigMixin):
         self.conv_out = InflatedConv3d(block_out_channels[0], out_channels, kernel_size=3, padding=1)
         
         self.skeleton_adapter = Adapter(cin=int(3*64), channels=[320, 640, 1280, 1280][:4], nums_rb=2, ksize=1, sk=True, use_conv=False)
-        adapter_weight = torch.load('./FollowYourPose/checkpoints/t2iadapter_keypose_sd14v1.pth')
+        adapter_weight = torch.load('./FollowYourPose/checkpoints/pose_encoder.pth')
         self.skeleton_adapter.load_state_dict(adapter_weight)
         
         

inference_followyourpose.py

@@ -43,7 +43,7 @@ class merge_config_then_run():
         bottom_crop=0,
     ):
         
-        default_edit_config='./FollowYourPose/configs/pose_sample.yaml'
+        default_edit_config='/home/user/app/FollowYourPose/FollowYourPose/configs/pose_sample.yaml'
         Omegadict_default_edit_config = OmegaConf.load(default_edit_config)
         
         dataset_time_string = get_time_string()