Voltline's picture
Release vimeml-tiny-zh-v1 (step 844730): FP32 bundle, INT8 Core ML packages, source snapshot (part 2)
99297d7 verified
Raw History Blame Contribute Delete
387 Bytes
"""Child process capturing SentencePiece's native C++ training diagnostics."""
import json
import sys
from pathlib import Path
def main():
import sentencepiece as spm
request = json.loads(Path(sys.argv[1]).read_text(encoding='utf-8'))
spm.set_random_generator_seed(request['seed'])
spm.SentencePieceTrainer.train(**request['options'])
if __name__=='__main__': main()