From 0de093a8766f71bd8a7a8b9ee45c42cd9382d589 Mon Sep 17 00:00:00 2001 From: Longtsing Date: Sat, 2 Nov 2024 22:16:05 +0800 Subject: [PATCH] Update diffusion model.py fix the error "name 'x_seqD' is not defined" --- Diffusion-Model/diffusion model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Diffusion-Model/diffusion model.py b/Diffusion-Model/diffusion model.py index 9c46851..b951d5c 100644 --- a/Diffusion-Model/diffusion model.py +++ b/Diffusion-Model/diffusion model.py @@ -159,7 +159,7 @@ def p_sample_loop(model, shape, n_steps, betas, one_minus_alphas_bar_sqrt): for i in reversed(range(n_steps)): cur_x = p_sample(model, cur_x, i, betas, one_minus_alphas_bar_sqrt) x_seq.append(cur_x) - return x_seqD + return x_seq def p_sample(model, x, t, betas, one_minus_alphas_bar_sqrt):