correct

2022-07-01 17:28:29 +00:00 · 2022-07-01 17:28:29 +00:00 · 221de0edee
parent 0eac7bd682
commit 221de0edee
1 changed files with 1 additions and 0 deletions
--- 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