Linoy Tsaban commited on
Commit
2822d34
1 Parent(s): e1cb97f

Update inversion_utils.py

Browse files
Files changed (1) hide show
  1. 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 = True,
134
  prompt = "",
135
  cfg_scale = 3.5,
136
  num_inference_steps=50, eps = None
137
- progress=gr.Progress()):
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 = True,
247
  zs = None,
248
  controller=None,
249
- asyrp = False,
250
- progress=gr.Progress()):
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