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])
|