load in advance

app.py

@@ -63,7 +63,7 @@ def infer(prompt, output_size, ddim_steps, guidance_scale, cosine_scale, seed, o
         elif output_size == "2048 x 1024":
             resolutions_list = [[1024, 512],
                                 [2048, 1024]]
-        restart_steps = [int(ddim_steps * 0.3)] * len(resolutions_list)
+        restart_steps = [int(ddim_steps * 0.3)]
 
         # print('GPU starts')
         result = infer_gpu_part(pipe, seed, prompt, negative_prompt, ddim_steps, guidance_scale, resolutions_list, fast_mode, cosine_scale, disable_freeu, restart_steps)
@@ -83,7 +83,7 @@ def infer(prompt, output_size, ddim_steps, guidance_scale, cosine_scale, seed, o
             resolutions_list = [[512, 256],
                                 [1024, 512],
                                 [2048, 1024]]
-        restart_steps = [int(ddim_steps * 0.5)] * len(resolutions_list)
+        restart_steps = [int(ddim_steps * 0.5), int(ddim_steps * 0.8)]
 
         # print('GPU starts')
         result = infer_gpu_part_turbo(pipe_turbo, seed, prompt, negative_prompt, ddim_steps, guidance_scale, resolutions_list, fast_mode, cosine_scale, disable_freeu, restart_steps)