Linoy Tsaban
commited on
Commit
·
257ea11
1
Parent(s):
a02e0e3
Update app.py
Browse files
app.py
CHANGED
|
@@ -153,15 +153,15 @@ def invert_and_reconstruct(
|
|
| 153 |
|
| 154 |
def edit(input_image,
|
| 155 |
wts, zs,
|
| 156 |
-
tar_prompt
|
| 157 |
-
steps
|
| 158 |
-
skip
|
| 159 |
-
tar_cfg_scale
|
| 160 |
-
|
| 161 |
-
|
| 162 |
-
|
| 163 |
-
|
| 164 |
-
|
| 165 |
|
| 166 |
):
|
| 167 |
|
|
@@ -171,11 +171,11 @@ def edit(input_image,
|
|
| 171 |
|
| 172 |
|
| 173 |
editing_args = dict(
|
| 174 |
-
editing_prompt =
|
| 175 |
-
reverse_editing_direction =
|
| 176 |
-
edit_warmup_steps=
|
| 177 |
-
edit_guidance_scale=
|
| 178 |
-
edit_threshold=
|
| 179 |
edit_momentum_scale=0.5,
|
| 180 |
edit_mom_beta=0.6,
|
| 181 |
eta=1,
|
|
@@ -343,11 +343,11 @@ with gr.Blocks(css='style.css') as demo:
|
|
| 343 |
steps,
|
| 344 |
skip,
|
| 345 |
tar_cfg_scale,
|
| 346 |
-
|
| 347 |
-
|
| 348 |
-
|
| 349 |
-
|
| 350 |
-
|
| 351 |
|
| 352 |
],
|
| 353 |
outputs=[sega_edited_image],
|
|
|
|
| 153 |
|
| 154 |
def edit(input_image,
|
| 155 |
wts, zs,
|
| 156 |
+
tar_prompt,
|
| 157 |
+
steps,
|
| 158 |
+
skipת
|
| 159 |
+
tar_cfg_scale,
|
| 160 |
+
edit_concept_1,edit_concept_2,edit_concept_3,
|
| 161 |
+
guidnace_scale_1,guidnace_scale_2,guidnace_scale_3,
|
| 162 |
+
warmup_1, warmup_2, warmup_3,
|
| 163 |
+
neg_guidance_1, neg_guidance_2, neg_guidance_3,
|
| 164 |
+
threshold_1, threshold_2, threshold_3
|
| 165 |
|
| 166 |
):
|
| 167 |
|
|
|
|
| 171 |
|
| 172 |
|
| 173 |
editing_args = dict(
|
| 174 |
+
editing_prompt = [edit_concept_1,edit_concept_2,edit_concept_3],
|
| 175 |
+
reverse_editing_direction = [ neg_guidance_1, neg_guidance_2, neg_guidance_3,],
|
| 176 |
+
edit_warmup_steps=[warmup_1, warmup_2, warmup_3,],
|
| 177 |
+
edit_guidance_scale=[guidnace_scale_1,guidnace_scale_2,guidnace_scale_3],
|
| 178 |
+
edit_threshold=[threshold_1, threshold_2, threshold_3],
|
| 179 |
edit_momentum_scale=0.5,
|
| 180 |
edit_mom_beta=0.6,
|
| 181 |
eta=1,
|
|
|
|
| 343 |
steps,
|
| 344 |
skip,
|
| 345 |
tar_cfg_scale,
|
| 346 |
+
edit_concept_1,edit_concept_2,edit_concept_3,
|
| 347 |
+
guidnace_scale_1,guidnace_scale_2,guidnace_scale_3,
|
| 348 |
+
warmup_1, warmup_2, warmup_3,
|
| 349 |
+
neg_guidance_1, neg_guidance_2, neg_guidance_3,
|
| 350 |
+
threshold_1, threshold_2, threshold_3
|
| 351 |
|
| 352 |
],
|
| 353 |
outputs=[sega_edited_image],
|