From 280984f43fc9b30eeeca3aae83054d87f487221b Mon Sep 17 00:00:00 2001 From: Jason Schleifer Date: Fri, 24 May 2024 11:33:15 -0700 Subject: [PATCH] using boolean instead of string for Leonardo --- nodes/image_drivers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nodes/image_drivers.py b/nodes/image_drivers.py index 32ac5cc..2110b1a 100644 --- a/nodes/image_drivers.py +++ b/nodes/image_drivers.py @@ -109,7 +109,7 @@ class gtUILeonardoImageGenerationDriver(gtUIBaseImageGenerationDriver): inputs["required"].update( { "model": (models, {"default": models[0]}), - "use_custom_model": (["yes", "no"], {"default": "no"}), + "use_custom_model": ("BOOLEAN", {"default": False}), "custom_model": ("STRING", {"default": ""}), } ) @@ -122,7 +122,7 @@ class gtUILeonardoImageGenerationDriver(gtUIBaseImageGenerationDriver): return None def create(self, model, prompt, use_custom_model=False, custom_model=""): - if use_custom_model == "yes" and custom_model != "": + if use_custom_model and custom_model != "": m = custom_model else: m = self.get_model_by_name(model)