Skip to content

Commit f9d988c

Browse files
authored
fix n_min mistake
1 parent cfaebb0 commit f9d988c

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

FlowEdit_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -214,11 +214,11 @@ def FlowEditSD3(pipe,
214214

215215
src_tar_latent_model_input = torch.cat([xt_tar, xt_tar, xt_tar, xt_tar]) if pipe.do_classifier_free_guidance else (xt_src, xt_tar)
216216

217-
_, noise_pred_tar = calc_v_sd3(pipe, src_tar_latent_model_input,src_tar_prompt_embeds, src_tar_pooled_prompt_embeds, src_guidance_scale, tar_guidance_scale, t)
217+
_, Vt_tar = calc_v_sd3(pipe, src_tar_latent_model_input,src_tar_prompt_embeds, src_tar_pooled_prompt_embeds, src_guidance_scale, tar_guidance_scale, t)
218218

219219
xt_tar = xt_tar.to(torch.float32)
220220

221-
prev_sample = xt_tar + (t_im1 - t_im1) * (noise_pred_tar)
221+
prev_sample = xt_tar + (t_im1 - t_i) * (Vt_tar)
222222

223223
prev_sample = prev_sample.to(noise_pred_tar.dtype)
224224

0 commit comments

Comments
 (0)