correct

221de0ed · Patrick von Platen · 0eac7bd6 · 221de0ed
Commit 221de0ed authored Jul 01, 2022 by Patrick von Platen
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

src/diffusers/models/unet_sde_score_estimation.py src/diffusers/models/unet_sde_score_estimation.py +1 -0

No files found.
--- a/src/diffusers/models/unet_sde_score_estimation.py
+++ b/src/diffusers/models/unet_sde_score_estimation.py
@@ -276,6 +276,7 @@ class NCSNpp(ModelMixin, ConfigMixin):
            skip_rescale=skip_rescale,
            continuous=continuous,
        )
+        self.act = nn.SiLU()
        self.nf = nf
        self.num_res_blocks = num_res_blocks
        self.attn_resolutions = attn_resolutions