fix model variants
Browse files
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 |
-
|
606 |
return fluxaudio_s(**kwargs)
|
607 |
-
|
608 |
-
|
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__':
|