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]: