fix: update to how to train

This commit is contained in:
Zach Nussbaum 2023-03-25 21:57:01 +00:00
parent 134d566b79
commit 2ac939529d

View File

@ -24,3 +24,8 @@ pip install -e .
## Generate
`python generate.py --config configs/generate/generate.yaml --prompt "Write a script to reverse a string in Python`
## Train
`accelerate launch --dynamo_backend=inductor --num_processes=8 --num_machines=1 --machine_rank=0 --deepspeed_multinode_launcher standard --mixed_precision=bf16 --use_deepspeed --deepspeed_config_file=configs/deepspeed/ds_config.json train.py --config configs/train/finetune-7b.yaml`