Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 =
|
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,
|