Spaces:
Sleeping
Sleeping
return noise to pipe_inference.
Browse files- app.py +1 -1
- 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 |
-
|
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,
|