AndreasXi commited on
Commit
ef27943
·
1 Parent(s): 98c6962

update meanaudio_l_full

Browse files
Files changed (1) hide show
  1. meanaudio/model/networks.py +2 -2
meanaudio/model/networks.py CHANGED
@@ -602,8 +602,8 @@ def meanaudio_l(**kwargs) -> MeanAudio:
602
  return MeanAudio(latent_dim=20,
603
  text_dim=1024,
604
  hidden_dim=64 * num_heads,
605
- depth=24,
606
- fused_depth=16,
607
  num_heads=num_heads,
608
  latent_seq_len=312, # for 10s audio
609
  **kwargs)
 
602
  return MeanAudio(latent_dim=20,
603
  text_dim=1024,
604
  hidden_dim=64 * num_heads,
605
+ depth=12,
606
+ fused_depth=8,
607
  num_heads=num_heads,
608
  latent_seq_len=312, # for 10s audio
609
  **kwargs)