Update README.md

README.md

@@ -14,18 +14,20 @@ You can use it like this:
 
 
 ```python
-  model = LlamaForCausalLM.from_pretrained(
-      "decapoda-research/llama-13b-hf",
-      load_in_8bit=LOAD_8BIT,
-      torch_dtype=torch.float16,
-      device_map="auto",
-  )
-  model = PeftModel.from_pretrained(
-      model,
-      "Chinese-Vicuna/Chinese-Vicuna-lora-13b-belle-and-guanaco",
-      torch_dtype=torch.float16,
-      device_map={'': 0}
-  )
+from transformers import LlamaForCausalLM
+from peft import PeftModel
 
+model = LlamaForCausalLM.from_pretrained(
+    "decapoda-research/llama-13b-hf",
+    load_in_8bit=True,
+    torch_dtype=torch.float16,
+    device_map="auto",
+)
+model = PeftModel.from_pretrained(
+    model,
+    LORA_PATH, # specific checkpoint path from "Chinese-Vicuna/Chinese-Vicuna-lora-13b-belle-and-guanaco"
+    torch_dtype=torch.float16,
+    device_map={'': 0}
+)
 ```