From 88922929cb36b95166e64f9b0d7f0626a2688d75 Mon Sep 17 00:00:00 2001 From: wassname Date: Thu, 22 Dec 2022 11:01:37 +0800 Subject: [PATCH] Update classifier_free_guidance.py --- denoising_diffusion_pytorch/classifier_free_guidance.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/denoising_diffusion_pytorch/classifier_free_guidance.py b/denoising_diffusion_pytorch/classifier_free_guidance.py index cc941fd..28a20ed 100644 --- a/denoising_diffusion_pytorch/classifier_free_guidance.py +++ b/denoising_diffusion_pytorch/classifier_free_guidance.py @@ -488,7 +488,7 @@ class GaussianDiffusion(nn.Module): beta_schedule = 'cosine', p2_loss_weight_gamma = 0., # p2 loss weight, from https://arxiv.org/abs/2204.00227 - 0 is equivalent to weight of 1 across time - 1. is recommended p2_loss_weight_k = 1, - ddim_sampling_eta = 1. + ddim_sampling_eta = 0. ): super().__init__() assert not (type(self) == GaussianDiffusion and model.channels != model.out_dim)