akshay107
/

llama-2-13b-minecraft-actseq-adapter

Model card Files Files and versions Community

akshay107 commited on Mar 26

Commit

37fb9c7

•

1 Parent(s): db460e7

Update merge.py

Files changed (1) hide show

merge.py +0 -1

merge.py CHANGED Viewed

@@ -7,7 +7,6 @@ from peft import LoraConfig, PeftModel
 device_map = "auto"
 model = AutoModelForCausalLM.from_pretrained(
     "/path/to/meta-llama2-13b",
-    #low_cpu_mem_usage=True,
     return_dict=True,
     torch_dtype=torch.float16,
     device_map=device_map,

 device_map = "auto"
 model = AutoModelForCausalLM.from_pretrained(
     "/path/to/meta-llama2-13b",
     return_dict=True,
     torch_dtype=torch.float16,
     device_map=device_map,