Typo switch width x height

lucataco · Nov 21, 2023 · 9b7e960 · 9b7e960
1 parent 9c3ea36
commit 9b7e960
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/predict.py b/predict.py
@@ -63,7 +63,7 @@ def predict(
             'save_path': "/src/results/",
             'model': 'UNet',
             'num_frames': num_frames,
-            'image_size': [width, height],
+            'image_size': [height, width],
             'use_fp16': True,
             'enable_xformers_memory_efficient_attention': True,
             'seed': str(seed),