Anonymous commited on
Commit
21f92ab
·
1 Parent(s): a266b99

load in advance

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -83,7 +83,7 @@ def infer(prompt, output_size, ddim_steps, guidance_scale, cosine_scale, seed, o
83
  resolutions_list = [[512, 256],
84
  [1024, 512],
85
  [2048, 1024]]
86
- restart_steps = [int(ddim_steps * 0.5), int(ddim_steps * 0.75)]
87
 
88
  # print('GPU starts')
89
  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)
 
83
  resolutions_list = [[512, 256],
84
  [1024, 512],
85
  [2048, 1024]]
86
+ restart_steps = [int(ddim_steps * 0.5)] * 2
87
 
88
  # print('GPU starts')
89
  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)