Spaces:

MoonQiu
/

FreeScale

Running on Zero

Anonymous commited on Dec 20, 2024

Commit

6f34b78

1 Parent(s): f91119a

load in advance

Files changed (1) hide show

app.py CHANGED Viewed

@@ -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)
         # 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)
         # print('GPU ends')
     return result

         restart_steps = [int(ddim_steps * 0.3)] * 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)
         # print('GPU ends')
     else:
         restart_steps = [int(ddim_steps * 0.5)] * 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)
         # print('GPU ends')
     return result