using config for device selection

app.py

@@ -52,6 +52,7 @@ model_configs = appConfig.get("models", {})
 models = list(model_configs.keys())
 scheduler_configs = appConfig.get("schedulers", {})
 schedulers = list(scheduler_configs.keys())
+devices =  appConfig.get("devices", [])
 
 device = None
 variant = None
@@ -266,7 +267,7 @@ with gr.Blocks() as demo:
     gr.Markdown("## Image Generation")
     gr.Markdown("### Device specific settings")
     with gr.Row():
-        rg_device = gr.Radio(label="Device:", value=device, choices=["cpu", "mps", "gpu"])
+        el_devices = gr.Dropdown(label="Device:", value=device, choices=devices, filterable=True, multiselect=False, allow_custom_value=True)
         rg_data_type = gr.Radio(label="Data Type:", value=data_type, choices=["bfloat16", "float16"], info="`blfoat16` is not supported on MPS devices right now; Half-precision weights, will save GPU memory, see https://huggingface.co/docs/diffusers/main/en/optimization/fp16")
         rg_allow_tensorfloat32 = gr.Radio(label="Allow TensorFloat32:", value=allow_tensorfloat32, choices=[True, False], info="is not supported on MPS devices right now; use TensorFloat-32 is faster, but results in slightly less accurate computations, see https://huggingface.co/docs/diffusers/main/en/optimization/fp16 ")
         rg_variant = gr.Radio(label="Variant:", value=variant, choices=["fp16", None], info="Use half-precision weights will save GPU memory, not all models support that, see https://huggingface.co/docs/diffusers/main/en/optimization/fp16 ")
@@ -307,7 +308,7 @@ with gr.Blocks() as demo:
         el_result = gr.Image()
         txt_code = gr.Code(get_sorted_code(), label="Code")
     
-    rg_device.change(device_change, inputs=[rg_device], outputs=[txt_code])
+    el_devices.change(device_change, inputs=[el_devices], outputs=[txt_code])
     rg_data_type.change(data_type_change, inputs=[rg_data_type], outputs=[txt_code])
     rg_allow_tensorfloat32.change(tensorfloat32_change, inputs=[rg_allow_tensorfloat32], outputs=[txt_code])
     rg_variant.change(variant_change, inputs=[rg_variant], outputs=[txt_code])
@@ -317,7 +318,7 @@ with gr.Blocks() as demo:
     dd_schedulers.change(schedulers_change, inputs=[dd_schedulers], outputs=[txt_code, txt_scheduler])
     btn_start_pipeline.click(start_pipeline, inputs=[
         dd_models, 
-        rg_device,
+        el_devices,
         rg_use_safetensors,
         rg_data_type,
         rg_variant,

appConfig.json

@@ -11,6 +11,9 @@
         }
 
     },
+    "devices": [
+        "cpu", "cuda", "ipu", "xpu", "mkldnn", "opengl", "opencl", "ideep", "hip", "ve", "fpga", "ort", "xla", "lazy", "vulkan", "mps", "meta", "hpu", "mtia", "privateuseone", "gpu"
+    ],
     "schedulers": {
         "DDPMScheduler":                    "This is DDPM",  
         "DDIMScheduler":                    "This is DDIM",