From 3d03901f6e3128f4139c864d616e3bc4dcfa6c01 Mon Sep 17 00:00:00 2001
From: Sakib Ahamed <35700378+zsxkib@users.noreply.github.com>
Date: Mon, 9 Sep 2024 23:03:13 +0200
Subject: [PATCH] Fix/aspect ratios (#30)

* Fix: aspect ratios now work correctly in predict.py
---
 predict.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/predict.py b/predict.py
index 7aee156..78a66aa 100644
--- 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}