Prgckwb commited on
Commit
59b2b05
1 Parent(s): 2075cf0

:tada: init

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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])