tipi/kyutai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

More MLX VAD example.

Browse Source
This commit is contained in:
Laurent 2025-07-08 15:59:42 +02:00
parent dc6f552a47
commit e76b2a53ff
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
scripts/stt_from_file_mlx.py
View File

@ -48,7 +48,10 @@ if __name__ == "__main__":
nn.quantize(model, bits=8, group_size=64)
print(f"loading model weights from {moshi_weights}")
model.load_weights(moshi_weights, strict=True)
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}")
text_tokenizer = sentencepiece.SentencePieceProcessor(text_tokenizer) # type: ignore