storytold/storyteller-ml
2
0
Fork 0
mirror of https://github.com/storytold/storyteller-ml.git synced 2026-10-25 08:40:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix resolution issue

Browse Source
This commit is contained in:
Brandon Thomas
2025-02-04 21:56:21 -05:00
parent ebf873e4b1
commit 0110e2ea42
3 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
animation/animate-x/animatex/model/unet_animate_x.py
+2 -1
View File
@@ -237,7 +237,8 @@ class UNetSD_Animate_X(nn.Module):
self.video_compositions = ['image', 'local_image', 'dwpose', 'randomref', 'randomref_pose', 'pose_embedding'] self.video_compositions = ['image', 'local_image', 'dwpose', 'randomref', 'randomref_pose', 'pose_embedding']
self.resolution = [512, 768] #self.resolution = [512, 768]
self.resolution = config['resolution'] # NB(bt,2025-02-04): Required for variable resolutions
self.in_dim = in_dim self.in_dim = in_dim
animation/animate-x/inference_cli.py
+3
View File
@@ -59,6 +59,9 @@ def parse_args():
default=14, default=14,
required=False required=False
) )
# TODO(bt): Means to load model files
#parser.add_argument( #parser.add_argument(
# "--pretrained_model_name_or_path", # "--pretrained_model_name_or_path",
# type=str, # type=str,
animation/animate-x/run_inference.sh
+4 -1
View File
@@ -4,9 +4,12 @@
python inference_cli.py \ python inference_cli.py \
--cfg configs/Animate_X_infer.yaml \ --cfg configs/Animate_X_infer.yaml \
--image_file data_test_copy/images/astronaut.jpg \ --image_file data_test_copy/images/future_girl_2.png \
--pose_directory data_test_copy/saved_pose/dance_1 \ --pose_directory data_test_copy/saved_pose/dance_1 \
--frame_directory data_test_copy/saved_frames/dance_1 \ --frame_directory data_test_copy/saved_frames/dance_1 \
--pickle_data_file data_test_copy/saved_pkl/dance_1.pkl \ --pickle_data_file data_test_copy/saved_pkl/dance_1.pkl \
--width 768 \ --width 768 \
--height 512 --height 512
# --height 512
# --height 768