diff --git a/animation/animate-x/animatex/model/unet_animate_x.py b/animation/animate-x/animatex/model/unet_animate_x.py
index 6ecad91..4052cb1 100644
--- a/animation/animate-x/animatex/model/unet_animate_x.py
+++ b/animation/animate-x/animatex/model/unet_animate_x.py
@@ -237,7 +237,8 @@ class UNetSD_Animate_X(nn.Module):
 
 
         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
diff --git a/animation/animate-x/inference_cli.py b/animation/animate-x/inference_cli.py
index ba07625..c8315ac 100755
--- a/animation/animate-x/inference_cli.py
+++ b/animation/animate-x/inference_cli.py
@@ -59,6 +59,9 @@ def parse_args():
         default=14,
         required=False
     )
+
+    # TODO(bt): Means to load model files
+
     #parser.add_argument(
     #    "--pretrained_model_name_or_path",
     #    type=str,
diff --git a/animation/animate-x/run_inference.sh b/animation/animate-x/run_inference.sh
index ce30eef..1c623bb 100755
--- a/animation/animate-x/run_inference.sh
+++ b/animation/animate-x/run_inference.sh
@@ -4,9 +4,12 @@
 
 python inference_cli.py \
     --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 \
     --frame_directory data_test_copy/saved_frames/dance_1 \
     --pickle_data_file data_test_copy/saved_pkl/dance_1.pkl \
     --width 768 \
     --height 512
+
+#    --height 512
+#    --height 768