More MLX VAD example.

2025-07-08 15:59:42 +02:00 · 2025-07-08 15:59:42 +02:00 · e76b2a53ff
commit e76b2a53ff
parent dc6f552a47
1 changed files with 4 additions and 1 deletions
--- a/scripts/stt_from_file_mlx.py
+++ b/scripts/stt_from_file_mlx.py
@ -48,6 +48,9 @@ if __name__ == "__main__":
        nn.quantize(model, bits=8, group_size=64)

    print(f"loading model weights from {moshi_weights}")
+    if args.hf_repo.endswith("-candle"):
+        model.load_pytorch_weights(moshi_weights, lm_config, strict=True)
+    else:
        model.load_weights(moshi_weights, strict=True)

    print(f"loading the text tokenizer from {text_tokenizer}")