Update README.md

README.md

@@ -25,16 +25,21 @@ This model has been fine tuned with mosaicml/instruct-v3 dataset with 2 epoch on
 ## How to use?
 from peft import PeftModel
 
-# load the base model
+#load the base model
+
 model_path = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
+
 tokenizer=AutoTokenizer.from_pretrained(model_path)
+
 model = AutoModelForCausalLM.from_pretrained(
     model_path,
     torch_dtype = torch.bfloat16,
     device_map = "auto",
     trust_remote_code = True
 )
+
 #load the adapter 
+
 model_peft = PeftModel.from_pretrained(model, "azam25/TinyLlama_instruct_generation")
 
 messages = [{
@@ -46,6 +51,7 @@ messages = [{
 }]
 
 def generate_response(message, model):
+
   prompt = tokenizer.apply_chat_template(messages, tokenize=False)
   encoded_input = tokenizer(prompt,  return_tensors="pt", add_special_tokens=True)
   model_inputs = encoded_input.to('cuda')