load in advance

app.py

@@ -66,7 +66,7 @@ def infer(prompt, output_size, ddim_steps, guidance_scale, cosine_scale, seed, o
         restart_steps = [int(ddim_steps * 0.3)] * len(resolutions_list)
 
         # 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)
+        result = infer_gpu_part(pipe, seed, prompt, negative_prompt, ddim_steps, guidance_scale, resolutions_list, fast_mode, cosine_scale, disable_freeu, restart_steps)
         # print('GPU ends')
 
     else:
@@ -86,7 +86,7 @@ def infer(prompt, output_size, ddim_steps, guidance_scale, cosine_scale, seed, o
         restart_steps = [int(ddim_steps * 0.5)] * len(resolutions_list)
 
         # 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)
+        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)
         # print('GPU ends')
 
     return result