File size: 263 Bytes
bd63939
bfb5a01
 
bd63939
 
 
1
2
3
4
5
6
_target_: transformers.LlamaForCausalLM.from_pretrained
pretrained_model_name_or_path: AILab-CVC/seed-llama-14b-sft
# pretrained_model_name_or_path: ${oc.env:PROJECT_ROOT}/pretrained/seed_llama_14b_sft
load_in_8bit: True
# device_map: auto
low_cpu_mem_usage: True