add audio

2026-01-16 01:15:22 +01:00
parent 81daf3f67d
commit a658911f98
19 changed files with 2335 additions and 54 deletions
--- a/mlx_video/generate.py
+++ b/mlx_video/generate.py
@@ -215,7 +215,7 @@ def generate_video(
        prompt = text_encoder.enhance_t2v(prompt, max_tokens=max_tokens, temperature=temperature, seed=seed, verbose=verbose)
        print(f"{Colors.DIM}Enhanced: {prompt[:150]}{'...' if len(prompt) > 150 else ''}{Colors.RESET}")

-    text_embeddings, _ = text_encoder(prompt)
+    text_embeddings, _ = text_encoder(prompt, return_audio_embeddings=False)
    mx.eval(text_embeddings)

    del text_encoder