Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -23,6 +23,7 @@ pipe.load_lora_weights(hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8
|
|
23 |
pipe.fuse_lora(lora_scale=0.125)
|
24 |
pipe.to(DEVICE)
|
25 |
|
|
|
26 |
def reset_do_inversion():
|
27 |
return True
|
28 |
|
@@ -109,14 +110,6 @@ Stochastic Rectified Differential Equations*](https://rf-inversion.github.io/dat
|
|
109 |
placeholder="describe the edited output",
|
110 |
)
|
111 |
with gr.Row():
|
112 |
-
eta = gr.Slider(
|
113 |
-
label="eta",
|
114 |
-
info = "lower eta to ehnace the edits",
|
115 |
-
minimum=0.0,
|
116 |
-
maximum=1.0,
|
117 |
-
step=0.1,
|
118 |
-
value=0.9,
|
119 |
-
)
|
120 |
start_timestep = gr.Slider(
|
121 |
label="start timestep",
|
122 |
info = "lower gamma to ehnace the edits",
|
@@ -133,6 +126,14 @@ Stochastic Rectified Differential Equations*](https://rf-inversion.github.io/dat
|
|
133 |
step=0.1,
|
134 |
value=0.9,
|
135 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
136 |
|
137 |
run_button = gr.Button("Edit", variant="primary")
|
138 |
|
@@ -140,16 +141,16 @@ Stochastic Rectified Differential Equations*](https://rf-inversion.github.io/dat
|
|
140 |
result = gr.Image(label="Result")
|
141 |
|
142 |
with gr.Accordion("Advanced Settings", open=False):
|
143 |
-
|
144 |
-
|
145 |
-
|
146 |
-
|
147 |
-
|
148 |
-
|
149 |
-
|
150 |
-
|
151 |
-
|
152 |
-
|
153 |
|
154 |
with gr.Row():
|
155 |
gamma = gr.Slider(
|
@@ -209,6 +210,13 @@ Stochastic Rectified Differential Equations*](https://rf-inversion.github.io/dat
|
|
209 |
outputs=[result, inverted_latents, image_latents, latent_image_ids, do_inversion, seed],
|
210 |
)
|
211 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
212 |
input_image.change(
|
213 |
fn=reset_do_inversion,
|
214 |
outputs=[do_inversion]
|
|
|
23 |
pipe.fuse_lora(lora_scale=0.125)
|
24 |
pipe.to(DEVICE)
|
25 |
|
26 |
+
examples = [[Image.open("cat.jpg"), "a tiger"]]
|
27 |
def reset_do_inversion():
|
28 |
return True
|
29 |
|
|
|
110 |
placeholder="describe the edited output",
|
111 |
)
|
112 |
with gr.Row():
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
113 |
start_timestep = gr.Slider(
|
114 |
label="start timestep",
|
115 |
info = "lower gamma to ehnace the edits",
|
|
|
126 |
step=0.1,
|
127 |
value=0.9,
|
128 |
)
|
129 |
+
eta = gr.Slider(
|
130 |
+
label="eta",
|
131 |
+
info = "lower eta to ehnace the edits",
|
132 |
+
minimum=0.0,
|
133 |
+
maximum=1.0,
|
134 |
+
step=0.1,
|
135 |
+
value=0.9,
|
136 |
+
)
|
137 |
|
138 |
run_button = gr.Button("Edit", variant="primary")
|
139 |
|
|
|
141 |
result = gr.Image(label="Result")
|
142 |
|
143 |
with gr.Accordion("Advanced Settings", open=False):
|
144 |
+
|
145 |
+
seed = gr.Slider(
|
146 |
+
label="Seed",
|
147 |
+
minimum=0,
|
148 |
+
maximum=MAX_SEED,
|
149 |
+
step=1,
|
150 |
+
value=42,
|
151 |
+
)
|
152 |
+
|
153 |
+
randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
|
154 |
|
155 |
with gr.Row():
|
156 |
gamma = gr.Slider(
|
|
|
210 |
outputs=[result, inverted_latents, image_latents, latent_image_ids, do_inversion, seed],
|
211 |
)
|
212 |
|
213 |
+
gr.Examples(
|
214 |
+
examples=examples,
|
215 |
+
inputs=[input_image, prompt,],
|
216 |
+
outputs=[result, inverted_latents, image_latents, latent_image_ids, do_inversion, seed],
|
217 |
+
fn=infer,
|
218 |
+
)
|
219 |
+
|
220 |
input_image.change(
|
221 |
fn=reset_do_inversion,
|
222 |
outputs=[do_inversion]
|