fix model variants
Browse files
meanaudio/model/networks.py
CHANGED
@@ -600,7 +600,7 @@ def meanaudio_s(**kwargs) -> MeanAudio:
|
|
600 |
|
601 |
|
602 |
def get_mean_audio(name: str, **kwargs) -> MeanAudio:
|
603 |
-
if name == 'meanaudio_s_ac' or 'meanaudio_s_full':
|
604 |
return meanaudio_s(**kwargs)
|
605 |
if name == 'fluxaudio_s_full':
|
606 |
return fluxaudio_s(**kwargs)
|
|
|
600 |
|
601 |
|
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)
|