AndreasXi commited on
Commit
7b916ea
·
1 Parent(s): 2b847cc

fix model variants

Browse files
Files changed (1) hide show
  1. meanaudio/model/networks.py +3 -3
meanaudio/model/networks.py CHANGED
@@ -602,10 +602,10 @@ def meanaudio_s(**kwargs) -> MeanAudio:
602
  def get_mean_audio(name: str, **kwargs) -> MeanAudio:
603
  if name == 'meanaudio_s_ac' or name == 'meanaudio_s_full':
604
  return meanaudio_s(**kwargs)
605
- if name == 'fluxaudio_s_full':
606
  return fluxaudio_s(**kwargs)
607
-
608
- raise ValueError(f'Unknown model name: {name}')
609
 
610
 
611
  if __name__ == '__main__':
 
602
  def get_mean_audio(name: str, **kwargs) -> MeanAudio:
603
  if name == 'meanaudio_s_ac' or name == 'meanaudio_s_full':
604
  return meanaudio_s(**kwargs)
605
+ elif name == 'fluxaudio_s_full':
606
  return fluxaudio_s(**kwargs)
607
+ else:
608
+ raise ValueError(f'Unknown model name: {name}')
609
 
610
 
611
  if __name__ == '__main__':