Barak1 commited on
Commit
c121b44
1 Parent(s): 4bf5a30

return noise to pipe_inference.

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. src/editor.py +1 -1
app.py CHANGED
@@ -147,7 +147,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
147
  @spaces.GPU
148
  def edit(inversion_state, target_prompt):
149
  if inversion_state is None:
150
- raise gr.Error("Set inputs before editing.")
151
  # if device == "cuda":
152
  # image = editor.to(device).edit(target_prompt)
153
  # else:
 
147
  @spaces.GPU
148
  def edit(inversion_state, target_prompt):
149
  if inversion_state is None:
150
+ raise gr.Error("Set inputs before editing.\nProgress indication below")
151
  # if device == "cuda":
152
  # image = editor.to(device).edit(target_prompt)
153
  # else:
src/editor.py CHANGED
@@ -77,7 +77,7 @@ class ImageEditorDemo:
77
  @staticmethod
78
  def edit(pipe_inference, target_prompt, last_latent, noise, cfg, edit_cfg):
79
  pipe_inference.cfg = cfg
80
- # pipe_inference.scheduler.set_noise_list(noise)
81
  pipe_inference.set_progress_bar_config(disable=True)
82
  image = pipe_inference(prompt=target_prompt,
83
  num_inference_steps=cfg.num_inference_steps,
 
77
  @staticmethod
78
  def edit(pipe_inference, target_prompt, last_latent, noise, cfg, edit_cfg):
79
  pipe_inference.cfg = cfg
80
+ pipe_inference.scheduler.set_noise_list(noise)
81
  pipe_inference.set_progress_bar_config(disable=True)
82
  image = pipe_inference(prompt=target_prompt,
83
  num_inference_steps=cfg.num_inference_steps,