Update classifier_free_guidance.py

2026-09-10 12:01:08 +08:00 · 2022-12-22 11:01:37 +08:00
1 changed files with 1 additions and 1 deletions
@@ -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)