mirror of
https://github.com/wassname/denoising-diffusion-pytorch.git
synced 2026-09-10 12:01:08 +08:00
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:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user