Update README.md

README.md

@@ -27,16 +27,19 @@ from peft import AutoPeftModelForCausalLM
 from transformers import AutoTokenizer, pipeline
 
 peft_model_id = "frankmorales2020/Mistral-7B-text-to-sql"
-# peft_model_id = args.output_dir
+
 
 # Load Model with PEFT adapter
+
 model = AutoPeftModelForCausalLM.from_pretrained(
   peft_model_id,
   device_map="auto",
   torch_dtype=torch.float16
 )
 tokenizer = AutoTokenizer.from_pretrained(peft_model_id)
+
 # Load into the pipeline
+
 pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)