fixed audiolm cannot run from local path bug

NexaAI · Dec 11, 2024 · c0d5fd4 · c0d5fd4
1 parent 33f6bab
commit c0d5fd4
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/nexa/cli/entry.py b/nexa/cli/entry.py
@@ -125,8 +125,12 @@ def run_ggml_inference(args):
             # from nexa.gguf.nexa_inference_tts import NexaTTSInference
             # inference = NexaTTSInference(model_path=model_path, local_path=local_path, **kwargs)
         elif run_type == "AudioLM":
-            from nexa.gguf.nexa_inference_audio_lm import NexaAudioLMInference
-            inference = NexaAudioLMInference(model_path=model_path, local_path=local_path, **kwargs)
+            if is_local_path:
+                from nexa.gguf.nexa_inference_audio_lm import NexaAudioLMInference
+                inference = NexaAudioLMInference(model_path=model_path, local_path=local_path, projector_local_path=projector_local_path, **kwargs)
+            else:
+                from nexa.gguf.nexa_inference_audio_lm import NexaAudioLMInference
+                inference = NexaAudioLMInference(model_path=model_path, local_path=local_path, **kwargs)
         else:
             print(f"Unknown task: {run_type}. Skipping inference.")
             return