From 1e35d0bcccc939aa409a2a087adb77e04f00e208 Mon Sep 17 00:00:00 2001 From: Ean Garvey Date: Fri, 31 May 2024 17:00:16 -0500 Subject: [PATCH] fixup conditional --- apps/shark_studio/api/sd.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/apps/shark_studio/api/sd.py b/apps/shark_studio/api/sd.py index 61f9b4591b..1b535b66b2 100644 --- a/apps/shark_studio/api/sd.py +++ b/apps/shark_studio/api/sd.py @@ -287,7 +287,7 @@ def shark_sd_fn_dict_input( sd_kwargs[key] = int(sd_kwargs[key]) # TODO: move these checks into the UI code so we don't have gradio warnings in a generalized dict input function. - if sd_kwargs["device"] == "": + if not sd_kwargs["device"]: gr.Warning("No device specified. Please specify a device.") return None, "" if sd_kwargs["height"] not in [512, 1024]: