Spaces:
Running
on
A10G
Running
on
A10G
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,
|