Skip to content

michal2409/LoRA

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

LLaMA2-70B LoRA finetuning

  1. Set MODEL env var to the LLaMa2-70B path in HF format.
export MODEL=<path>
  1. Build and run container
./docker_build.sh && ./docker_run.sh
  1. Prepare dataset
python3 prepare_dataset.py
  1. Convert model into nemo format
python NeMo/scripts/nlp_language_modeling/convert_hf_llama_to_nemo.py --in-file=/model/llama2-70b-hf/ --out-file=./model/llama2-70b.nemo
  1. Run training
./run.sh

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published