Default values

gradio_demo.py

@@ -156,6 +156,9 @@ def stage2_process(input_image, prompt, a_prompt, n_prompt, num_samples, upscale
 
 
 def load_and_reset(param_setting):
+    if torch.cuda.device_count() == 0:
+        gr.Warning('Set this space to GPU config to make it work.')
+        return None, None, None, None, None, None, None, None, None, None, None, None, None
     edm_steps = default_setting.edm_steps
     s_stage2 = 1.0
     s_stage1 = -1.0
@@ -250,9 +253,9 @@ with block:
                 num_samples = gr.Slider(label="Num Samples", minimum=1, maximum=4 if not args.use_image_slider else 1
                                         , value=1, step=1)
                 upscale = gr.Slider(label="Upscale", minimum=1, maximum=8, value=1, step=1)
-                edm_steps = gr.Slider(label="Steps", minimum=1, maximum=200, value=default_setting.edm_steps, step=1)
+                edm_steps = gr.Slider(label="Steps", minimum=1, maximum=200, value=default_setting.edm_steps if torch.cuda.device_count() > 0 else 1, step=1)
                 s_cfg = gr.Slider(label="Text Guidance Scale", minimum=1.0, maximum=15.0,
-                                  value=default_setting.s_cfg_Quality, step=0.1)
+                                  value=default_setting.s_cfg_Quality if torch.cuda.device_count() > 0 else 1.0, step=0.1)
                 s_stage2 = gr.Slider(label="Stage2 Guidance Strength", minimum=0., maximum=1., value=1., step=0.05)
                 s_stage1 = gr.Slider(label="Stage1 Guidance Strength", minimum=-1.0, maximum=6.0, value=-1.0, step=1.0)
                 seed = gr.Slider(label="Seed", minimum=-1, maximum=2147483647, step=1, randomize=True)
@@ -272,7 +275,7 @@ with block:
                     with gr.Column():
                         linear_CFG = gr.Checkbox(label="Linear CFG", value=True)
                         spt_linear_CFG = gr.Slider(label="CFG Start", minimum=1.0,
-                                                        maximum=9.0, value=default_setting.spt_linear_CFG_Quality, step=0.5)
+                                                        maximum=9.0, value=default_setting.spt_linear_CFG_Quality if torch.cuda.device_count() > 0 else 1.0, step=0.5)
                     with gr.Column():
                         linear_s_stage2 = gr.Checkbox(label="Linear Stage2 Guidance", value=False)
                         spt_linear_s_stage2 = gr.Slider(label="Guidance Start", minimum=0.,