Fix/aspect ratios (#30)

* Fix: aspect ratios now work correctly in predict.py
replicate · Sep 9, 2024 · 3d03901 · 3d03901
1 parent 8c881c9
commit 3d03901
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/predict.py b/predict.py
@@ -344,6 +344,8 @@ def predict(  # pyright: ignore
         else:  # is_txt2img_mode
             print("[!] txt2img mode")
             pipe = self.pipes[model]
+            flux_kwargs["width"] = width
+            flux_kwargs["height"] = height
 
         if replicate_weights:
             flux_kwargs["joint_attention_kwargs"] = {"scale": lora_scale}