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