torchrun --nproc_per_node 1 example_text_completionpy --ckpt_dir llama-2-7b --tokenizer_path tokenizermodel --max_seq_len 128 --max_batch_size 4将上述sh文件改为bat文件
torchrun --nproc_per_node 1 example_text_completion.py ^ --ckpt_dir llama-2-7b/ ^ --tokenizer_path tokenizer.model ^ --max_seq_len 128 --max_batch_size 4
原文地址: https://www.cveoy.top/t/topic/iLIt 著作权归作者所有。请勿转载和采集!