restore_in_9min

gradio_demo.py

@@ -146,37 +146,18 @@ def stage2_process(
     model_select,
     output_format
 ):
-    return restore(
-        noisy_image,
-        denoise_image,
-        prompt,
-        a_prompt,
-        n_prompt,
-        num_samples,
-        min_size,
-        downscale,
-        upscale,
-        edm_steps,
-        s_stage1,
-        s_stage2,
-        s_cfg,
-        randomize_seed,
-        seed,
-        s_churn,
-        s_noise,
-        color_fix_type,
-        diff_dtype,
-        ae_dtype,
-        gamma_correction,
-        linear_CFG,
-        linear_s_stage2,
-        spt_linear_CFG,
-        spt_linear_s_stage2,
-        model_select,
-        output_format
+    return restore_in_9min(
+        noisy_image, denoise_image, prompt, a_prompt, n_prompt, num_samples, min_size, downscale, upscale, edm_steps, s_stage1, s_stage2, s_cfg, randomize_seed, seed, s_churn, s_noise, color_fix_type, diff_dtype, ae_dtype, gamma_correction, linear_CFG, linear_s_stage2, spt_linear_CFG, spt_linear_s_stage2, model_select, output_format
     )
 
 @spaces.GPU(duration=540)
+def restore_in_9min(
+    noisy_image, denoise_image, prompt, a_prompt, n_prompt, num_samples, min_size, downscale, upscale, edm_steps, s_stage1, s_stage2, s_cfg, randomize_seed, seed, s_churn, s_noise, color_fix_type, diff_dtype, ae_dtype, gamma_correction, linear_CFG, linear_s_stage2, spt_linear_CFG, spt_linear_s_stage2, model_select, output_format
+):
+    return restore(
+        noisy_image, denoise_image, prompt, a_prompt, n_prompt, num_samples, min_size, downscale, upscale, edm_steps, s_stage1, s_stage2, s_cfg, randomize_seed, seed, s_churn, s_noise, color_fix_type, diff_dtype, ae_dtype, gamma_correction, linear_CFG, linear_s_stage2, spt_linear_CFG, spt_linear_s_stage2, model_select, output_format
+    )
+
 def restore(
     noisy_image,
     denoise_image,
@@ -285,13 +266,14 @@ def restore(
     hours = math.floor(minutes / 60)
     minutes = minutes - (hours * 60)
     information = ("Start the process again if you want a different result. " if randomize_seed else "") + \
-    "The new image resolution is " + str(result_width) + \
-    " pixels large and " + str(result_height) + \
-    " pixels high, so a resolution of " + f'{result_width * result_height:,}' + " pixels. " + \
+    "Wait 9 min before a new run to avoid time penalty. " + \
     "The image(s) has(ve) been generated in " + \
     ((str(hours) + " h, ") if hours != 0 else "") + \
     ((str(minutes) + " min, ") if hours != 0 or minutes != 0 else "") + \
-    str(secondes) + " sec."
+    str(secondes) + " sec. " + \
+    "The new image resolution is " + str(result_width) + \
+    " pixels large and " + str(result_height) + \
+    " pixels high, so a resolution of " + f'{result_width * result_height:,}' + " pixels."
     print(information)
 
     # Only one image can be shown in the slider