Update README.md

README.md

@@ -25,6 +25,7 @@ import torch
 from peft import PeftModel
 from transformers import AutoModelForCausalLM
 from transformers import LlamaTokenizer
+
 # Load base model and LoRA weights
 model = AutoModelForCausalLM.from_pretrained("haoranxu/ALMA-13B-Pretrain", torch_dtype=torch.float16, device_map="auto")
 model = PeftModel.from_pretrained(model, "haoranxu/ALMA-13B-Pretrain-LoRA")