Update README.md

README.md

@@ -38,12 +38,17 @@ The following `bitsandbytes` quantization config was used during training:
 ## Code to call this mnodel
 
 import torch
+
 from peft import PeftModel, PeftConfig
+
 from transformers import AutoModelForCausalLM, AutoTokenizer
+
 from transformers import BitsAndBytesConfig
 
 peft_model_id = "meetrais/finetuned_mistral_7b"
+
 config = PeftConfig.from_pretrained(peft_model_id)
+
 bnb_config = BitsAndBytesConfig(
 load_in_4bit=True,
 bnb_4bit_use_double_quant=True,
@@ -51,6 +56,7 @@ bnb_4bit_quant_type="nf4",
 bnb_4bit_compute_dtype=torch.bfloat16
 )
 model = AutoModelForCausalLM.from_pretrained(peft_model_id,  quantization_config=bnb_config, device_map='auto')
+
 tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
 
 if tokenizer.pad_token is None: