fix model variants

app.py

@@ -132,7 +132,7 @@ def generate_audio_gradio(
     current_time_string = datetime.now().strftime("%Y%m%d_%H%M%S_%f")
     filename = f"{safe_prompt}_{current_time_string}.flac"
     save_path = OUTPUT_DIR / filename
-    torchaudio.save(str(save_path), audio, temp_seq_cfg.sampling_rate)
+    torchaudio.save(str(save_path), audio, seq_cfg.sampling_rate)
     log.info(f"Audio saved to {save_path}")
 
     gc.collect()

meanaudio/model/networks.py

@@ -600,9 +600,9 @@ def meanaudio_s(**kwargs) -> MeanAudio:
 
 
 def get_mean_audio(name: str, **kwargs) -> MeanAudio:
-    if name == 'meanaudio_s':
+    if name == 'meanaudio_s_ac' or 'meanaudio_s_full':
         return meanaudio_s(**kwargs)
-    if name == 'fluxaudio_s': 
+    if name == 'fluxaudio_s_full': 
         return fluxaudio_s(**kwargs)
 
     raise ValueError(f'Unknown model name: {name}')