소스 검색

fixed arg names

Hamid Shojanazeri 1 년 전
부모
커밋
569f8b7976
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 2 2
      docs/inference.md

+ 2 - 2
docs/inference.md

@@ -46,9 +46,9 @@ torchrun --nnodes 1 --nproc_per_node 8  llama_finetuning.py --enable_fsdp --mode
 ```
 Then convert your FSDP checkpoint to HuggingFace checkpoints using:
 ```bash
- python checkpoint_converter_fsdp_hf.py --model_name  PATH/to/FSDP/Checkpoints --save_dir PATH/to/save/checkpoints --model_path PATH/or/HF/model_name
+ python checkpoint_converter_fsdp_hf.py --fsdp_checkpoint_path  PATH/to/FSDP/Checkpoints --consolidated_model_path PATH/to/save/checkpoints --HF_model_path PATH/or/HF/model_name
 
- # --model_path specifies the HF Llama model name or path where it has config.json and tokenizer.json
+ # --HF_model_path specifies the HF Llama model name or path where it has config.json and tokenizer.json
  ```
 
 Then run inference using: