Skip to content

GitLab

Commit 221de0ed authored by Patrick von Platen's avatar Patrick von Platen
Browse files

correct

parent 0eac7bd6
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/diffusers/models/unet_sde_score_estimation.py
View file @ 221de0ed
...@@ -276,6 +276,7 @@ class NCSNpp(ModelMixin, ConfigMixin): ...@@ -276,6 +276,7 @@ class NCSNpp(ModelMixin, ConfigMixin):
skip_rescale=skip_rescale, skip_rescale=skip_rescale,
continuous=continuous, continuous=continuous,
) )
self.act = nn.SiLU()
self.nf = nf self.nf = nf
self.num_res_blocks = num_res_blocks self.num_res_blocks = num_res_blocks
self.attn_resolutions = attn_resolutions self.attn_resolutions = attn_resolutions
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment