Tony Lian commited on
Commit
7b0a7ad
β€’
1 Parent(s): 2335a8f

Fix scaling in input latents_bg

Browse files
Files changed (1) hide show
  1. utils/latents.py +2 -0
utils/latents.py CHANGED
@@ -147,5 +147,7 @@ def get_input_latents_list(model_dict, bg_seed, fg_seed_start, fg_blending_ratio
147
 
148
  input_latents_list.append(input_latents)
149
 
 
 
150
  return input_latents_list, latents_bg
151
 
 
147
 
148
  input_latents_list.append(input_latents)
149
 
150
+ latents_bg = latents_bg * scheduler.init_noise_sigma
151
+
152
  return input_latents_list, latents_bg
153