height dim correctly set in example (#289)

xdit-project · Sep 25, 2024 · db920b5 · db920b5
1 parent 9675e39
commit db920b5
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/examples/flux_example.py b/examples/flux_example.py
@@ -38,7 +38,7 @@ def main():
     start_time = time.time()
     output = pipe(
         height=input_config.height,
-        width=input_config.height,
+        width=input_config.width,
         prompt=input_config.prompt,
         num_inference_steps=input_config.num_inference_steps,
         output_type=input_config.output_type,

diff --git a/examples/hunyuandit_example.py b/examples/hunyuandit_example.py
@@ -30,7 +30,7 @@ def main():
     start_time = time.time()
     output = pipe(
         height=input_config.height,
-        width=input_config.height,
+        width=input_config.width,
         prompt=input_config.prompt,
         num_inference_steps=input_config.num_inference_steps,
         output_type=input_config.output_type,

diff --git a/examples/pixartalpha_example.py b/examples/pixartalpha_example.py
@@ -30,7 +30,7 @@ def main():
     start_time = time.time()
     output = pipe(
         height=input_config.height,
-        width=input_config.height,
+        width=input_config.width,
         prompt=input_config.prompt,
         num_inference_steps=input_config.num_inference_steps,
         output_type=input_config.output_type,

diff --git a/examples/pixartsigma_example.py b/examples/pixartsigma_example.py
@@ -30,7 +30,7 @@ def main():
     start_time = time.time()
     output = pipe(
         height=input_config.height,
-        width=input_config.height,
+        width=input_config.width,
         prompt=input_config.prompt,
         num_inference_steps=input_config.num_inference_steps,
         output_type=input_config.output_type,

diff --git a/examples/sd3_example.py b/examples/sd3_example.py
@@ -30,7 +30,7 @@ def main():
     start_time = time.time()
     output = pipe(
         height=input_config.height,
-        width=input_config.height,
+        width=input_config.width,
         prompt=input_config.prompt,
         num_inference_steps=input_config.num_inference_steps,
         output_type=input_config.output_type,