Compare commits

..
1 Commits
3 changed files with 4 additions and 4 deletions
@@ -488,7 +488,7 @@ class GaussianDiffusion(nn.Module):
beta_schedule = 'cosine', 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_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, p2_loss_weight_k = 1,
ddim_sampling_eta = 0. ddim_sampling_eta = 1.
): ):
super().__init__() super().__init__()
assert not (type(self) == GaussianDiffusion and model.channels != model.out_dim) assert not (type(self) == GaussianDiffusion and model.channels != model.out_dim)
+1 -1
View File
@@ -1 +1 @@
__version__ = '0.1.5' __version__ = '1.16.6'