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
@@ -52,7 +52,7 @@ class ElucidatedDiffusion(nn.Module):
S_noise = 1.003,
):
super().__init__()
assert net.learned_sinusoidal_cond
assert net.random_or_learned_sinusoidal_cond
self.self_condition = net.self_condition
self.net = net