From 554ee17de54b95432edd4465a65e75d809b4564f Mon Sep 17 00:00:00 2001 From: Zhang Peiyuan Date: Thu, 27 Feb 2025 16:02:44 -0800 Subject: [PATCH] [BUG] update cfg bug? (#223) --- fastvideo/distill.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fastvideo/distill.py b/fastvideo/distill.py index 488859ef9..4b4d67dee 100644 --- a/fastvideo/distill.py +++ b/fastvideo/distill.py @@ -160,7 +160,7 @@ def distill_one_step( uncond_prompt_mask.unsqueeze(0).expand(bsz, -1), return_dict=False, )[0].float() - teacher_output = cond_teacher_output + w * (cond_teacher_output - uncond_teacher_output) + teacher_output = uncond_teacher_output + w * (cond_teacher_output - uncond_teacher_output) x_prev = solver.euler_step(noisy_model_input, teacher_output, index) # 20.4.12. Get target LCM prediction on x_prev, w, c, t_n