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,
|