From 221de0edee129f9601857c7b7534d2a872382d7c Mon Sep 17 00:00:00 2001
From: Patrick von Platen <patrick.v.platen@gmail.com>
Date: Fri, 1 Jul 2022 17:28:29 +0000
Subject: [PATCH] correct

---
 src/diffusers/models/unet_sde_score_estimation.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/diffusers/models/unet_sde_score_estimation.py b/src/diffusers/models/unet_sde_score_estimation.py
index a4cdc8bd..f4c7f66c 100644
--- 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