Spaces:
Runtime error
Runtime error
Prgckwb
commited on
Commit
•
59b2b05
1
Parent(s):
2075cf0
:tada: init
Browse files
app.py
CHANGED
@@ -36,6 +36,8 @@ def inference(
|
|
36 |
per_step_time: int = 1,
|
37 |
n_total_steps: int = 1000,
|
38 |
):
|
|
|
|
|
39 |
scheduler = SCHEDULERS[scheduler_name].from_pretrained(MODEL_IDS[model_name], subfolder='scheduler')
|
40 |
scheduler.set_timesteps(num_inference_steps=n_total_steps)
|
41 |
timesteps = torch.flip(scheduler.timesteps, dims=[0])
|
|
|
36 |
per_step_time: int = 1,
|
37 |
n_total_steps: int = 1000,
|
38 |
):
|
39 |
+
yield image_pil, ""
|
40 |
+
|
41 |
scheduler = SCHEDULERS[scheduler_name].from_pretrained(MODEL_IDS[model_name], subfolder='scheduler')
|
42 |
scheduler.set_timesteps(num_inference_steps=n_total_steps)
|
43 |
timesteps = torch.flip(scheduler.timesteps, dims=[0])
|