Linoy Tsaban
commited on
Commit
·
2822d34
1
Parent(s):
e1cb97f
Update inversion_utils.py
Browse files- inversion_utils.py +5 -5
inversion_utils.py
CHANGED
|
@@ -130,11 +130,11 @@ def get_variance(model, timestep): #, prev_timestep):
|
|
| 130 |
|
| 131 |
def inversion_forward_process(model, x0,
|
| 132 |
etas = None,
|
| 133 |
-
prog_bar =
|
| 134 |
prompt = "",
|
| 135 |
cfg_scale = 3.5,
|
| 136 |
num_inference_steps=50, eps = None
|
| 137 |
-
|
| 138 |
|
| 139 |
if not prompt=="":
|
| 140 |
text_embeddings = encode_text(model, prompt)
|
|
@@ -243,11 +243,11 @@ def inversion_reverse_process(model,
|
|
| 243 |
etas = 0,
|
| 244 |
prompts = "",
|
| 245 |
cfg_scales = None,
|
| 246 |
-
prog_bar =
|
| 247 |
zs = None,
|
| 248 |
controller=None,
|
| 249 |
-
asyrp = False
|
| 250 |
-
|
| 251 |
|
| 252 |
batch_size = len(prompts)
|
| 253 |
|
|
|
|
| 130 |
|
| 131 |
def inversion_forward_process(model, x0,
|
| 132 |
etas = None,
|
| 133 |
+
prog_bar = False,
|
| 134 |
prompt = "",
|
| 135 |
cfg_scale = 3.5,
|
| 136 |
num_inference_steps=50, eps = None
|
| 137 |
+
):
|
| 138 |
|
| 139 |
if not prompt=="":
|
| 140 |
text_embeddings = encode_text(model, prompt)
|
|
|
|
| 243 |
etas = 0,
|
| 244 |
prompts = "",
|
| 245 |
cfg_scales = None,
|
| 246 |
+
prog_bar = False,
|
| 247 |
zs = None,
|
| 248 |
controller=None,
|
| 249 |
+
asyrp = False
|
| 250 |
+
):
|
| 251 |
|
| 252 |
batch_size = len(prompts)
|
| 253 |
|