add option for random fourier features, given misinterpretation of Katherine's code, thanks to @tmabraham for addressing this in https://github.com/lucidrains/denoising-diffusion-pytorch/issues/112

This commit is contained in:
Phil Wang
2022-10-30 09:32:06 -07:00
parent aadaa7d288
commit 0ffff59ca0
5 changed files with 13 additions and 12 deletions
@@ -126,7 +126,7 @@ class ContinuousTimeGaussianDiffusion(nn.Module):
p2_loss_weight_k = 1
):
super().__init__()
assert model.learned_sinusoidal_cond
assert model.random_or_learned_sinusoidal_cond
assert not model.self_condition, 'not supported yet'
self.model = model