LPX55 commited on
Commit
8a23c91
·
verified ·
1 Parent(s): 889c6e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -217,7 +217,7 @@ def call_pipe(
217
 
218
  # 4. Prepare timesteps
219
  sigmas = np.linspace(1.0, 0.0, num_inference_steps + 1)[:-1] if sigmas is None else sigmas
220
- timesteps, num_inference_steps = retrieve_argument timesteps(
221
  pipe.scheduler,
222
  num_inference_steps,
223
  device,
 
217
 
218
  # 4. Prepare timesteps
219
  sigmas = np.linspace(1.0, 0.0, num_inference_steps + 1)[:-1] if sigmas is None else sigmas
220
+ timesteps, num_inference_steps = retrieve_timesteps(
221
  pipe.scheduler,
222
  num_inference_steps,
223
  device,