Fix height-width

h2oai · Sep 20, 2024 · c823024 · c823024
1 parent 73fd35f
commit c823024
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 11 deletions.
diff --git a/src/version.py b/src/version.py
@@ -1 +1 @@
-__version__ = "4dc90dad603445139085c7da6bd79a14652aaa70"
+__version__ = "73fd35fc5773c2b1dcf36ea25262c6c61bbc1fa8"
diff --git a/src/vision/flux.py b/src/vision/flux.py
@@ -58,8 +58,8 @@ def make_image(prompt, filename=None, gpu_id='auto', pipe=None,
     makedirs(os.path.dirname(lock_file))  # ensure made
     with filelock.FileLock(lock_file):
         image = pipe(prompt=prompt,
-                     height=image_size.lower().split('x')[0],
-                     width=image_size.lower().split('x')[1],
+                     height=int(image_size.lower().split('x')[0]),
+                     width=int(image_size.lower().split('x')[1]),
                      num_inference_steps=image_num_inference_steps,
                      max_sequence_length=max_sequence_length,
                      guidance_scale=image_guidance_scale).images[0]

diff --git a/src/vision/playv2.py b/src/vision/playv2.py
@@ -51,8 +51,8 @@ def make_image(prompt, filename=None, gpu_id='auto', pipe=None,
     makedirs(os.path.dirname(lock_file))  # ensure made
     with filelock.FileLock(lock_file):
         image = pipe(prompt=prompt,
-                     height=image_size.lower().split('x')[0],
-                     width=image_size.lower().split('x')[1],
+                     height=int(image_size.lower().split('x')[0]),
+                     width=int(image_size.lower().split('x')[1]),
                      num_inference_steps=image_num_inference_steps,
                      max_sequence_length=max_sequence_length,
                      guidance_scale=image_guidance_scale,

diff --git a/src/vision/sdxl_turbo.py b/src/vision/sdxl_turbo.py
@@ -52,8 +52,8 @@ def make_image(prompt, filename=None, gpu_id='auto', pipe=None,
     makedirs(os.path.dirname(lock_file))  # ensure made
     with filelock.FileLock(lock_file):
         image = pipe(prompt=prompt,
-                     height=image_size.lower().split('x')[0],
-                     width=image_size.lower().split('x')[1],
+                     height=int(image_size.lower().split('x')[0]),
+                     width=int(image_size.lower().split('x')[1]),
                      num_inference_steps=image_num_inference_steps,  # more than 1 not really helpful
                      guidance_scale=0.0,  # disabled: https://huggingface.co/stabilityai/sdxl-turbo#diffusers
                      ).images[0]

diff --git a/src/vision/stable_diffusion_xl.py b/src/vision/stable_diffusion_xl.py
@@ -93,17 +93,17 @@ def make_image(prompt,
         # run both experts
         image = base(
             prompt=prompt,
-            height=image_size.lower().split('x')[0],
-            width=image_size.lower().split('x')[1],
+            height=int(image_size.lower().split('x')[0]),
+            width=int(image_size.lower().split('x')[1]),
             num_inference_steps=image_num_inference_steps,
             guidance_scale=image_guidance_scale,
             **extra1,
         ).images
         if refiner:
             image = refiner(
                 prompt=prompt,
-                height=image_size.lower().split('x')[0],
-                width=image_size.lower().split('x')[1],
+                height=int(image_size.lower().split('x')[0]),
+                width=int(image_size.lower().split('x')[1]),
                 num_inference_steps=image_num_inference_steps,
                 guidance_scale=image_guidance_scale,
                 **extra2,