rust/delayed-streams-modeling
1
0
Fork 0
mirror of https://github.com/kyutai-labs/delayed-streams-modeling.git synced 2026-01-09 02:32:53 +00:00
Code Issues 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
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