Update classifier_free_guidance.py

0.1.5
always set results_folder on Trainer
2026-09-10 12:01:08 +08:00 · 2022-12-22 11:01:37 +08:00 · 2022-12-21 08:35:59 -08:00 · 2022-12-21 08:34:34 -08:00 · 2022-11-28 12:11:11 -08:00
3 changed files with 4 additions and 4 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)
@@ -810,8 +810,8 @@ class Trainer(object):
        if self.accelerator.is_main_process:
            self.ema = EMA(diffusion_model, beta = ema_decay, update_every = ema_update_every)

-            self.results_folder = Path(results_folder)
-            self.results_folder.mkdir(exist_ok = True)
+        self.results_folder = Path(results_folder)
+        self.results_folder.mkdir(exist_ok = True)

        # step counter state

@@ -1 +1 @@
-__version__ = '1.16.6'
+__version__ = '0.1.5'
Author	SHA1	Message	Date
wassname	88922929cb	Update classifier_free_guidance.py	2022-12-22 11:01:37 +08:00
Phil Wang	cf4f44909f	0.1.5	2022-12-21 08:35:59 -08:00
Phil Wang	a7f2d670bb	always set results_folder on Trainer	2022-12-21 08:34:34 -08:00
Phil Wang	6ab29d5cea	start saving version of library in the checkpoints, so in case of breaking changes, researchers can return to the right working version	2022-11-28 12:11:11 -08:00