mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

undo sandwich

Browse Source
This commit is contained in:
James Betker 2022-03-22 11:43:24 -06:00
parent 927731f3b4
commit f3f391b372
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
codes/models/audio/tts/unet_diffusion_tts_flat.py
View File

@ -133,7 +133,6 @@ class DiffusionTtsFlat(nn.Module):
rotary_pos_emb=True, rotary_pos_emb=True,
layerdrop_percent=layer_drop, layerdrop_percent=layer_drop,
zero_init_branch_output=True, zero_init_branch_output=True,
sandwich_coef=4,
) )
) )
self.layers.transformer.norm = nn.Identity() # We don't want the final norm for the main encoder. self.layers.transformer.norm = nn.Identity() # We don't want the final norm for the main encoder.