Update README.md

README.md

@@ -44,8 +44,11 @@ To use these weights:
 from peft import PeftModel, PeftConfig
 from transformers import AutoModelForCausalLM
 
-config = PeftConfig.from_pretrained("arnavgrg/codealpaca-qlora", load_in_4bit=True)
-model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-7b-hf")
+# Load base model in 4 bit
+model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-7b-hf", load_in_4bit=True)
+
+# Wrap model with pretrained model weights
+config = PeftConfig.from_pretrained("arnavgrg/codealpaca-qlora")
 model = PeftModel.from_pretrained(model, "arnavgrg/codealpaca-qlora")
 ```