diff --git a/src/version.py b/src/version.py index edc1160e1..85a5966e9 100644 --- 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 index 4e2ab6e9d..24e8efdea 100644 --- 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 index 2319ab97f..53daa4653 100644 --- 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 index 1ea921854..272532070 100644 --- 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 index cba7a4fda..1e92219c9 100644 --- a/src/vision/stable_diffusion_xl.py +++ b/src/vision/stable_diffusion_xl.py @@ -93,8 +93,8 @@ 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, @@ -102,8 +102,8 @@ def make_image(prompt, 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,