Mirrors
/
diffusers
mirror of https://github.com/ShivamShrirao/diffusers.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

correct

This commit is contained in:
Patrick von Platen 2022-07-01 17:28:29 +00:00
parent 0eac7bd682
commit 221de0edee
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/diffusers/models/unet_sde_score_estimation.py
View File

@ -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