Update README.md

README.md

@@ -17,7 +17,7 @@ widget:
 ---
 
 # LlaMa2-CodeGen
-This model is **LlaMa-2 7b** [Llama-2](https://huggingface.co/meta-llama/Llama-2-7b) fine-tuned on the **CodeSearchNet dataset** [CodeSearchNet](https://github.com/github/CodeSearchNet) by using the method **QLoRA** with [PEFT](https://github.com/huggingface/peft) library.
+This model is **LlaMa-2 7b** [Llama-2](https://huggingface.co/meta-llama/Llama-2-7b) fine-tuned on the **CodeSearchNet dataset** [CodeSearchNet](https://github.com/github/CodeSearchNet) by using the method **QLoRA** [QLoRA](https://github.com/artidoro/qlora) with [PEFT](https://github.com/huggingface/peft) library.
 
 # Model Trained on Google Colab Pro Using AutoTrain, PEFT and QLoRA
 
@@ -31,7 +31,7 @@ import torch
 from peft import PeftModel, PeftConfig
 from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
 
-peft_model_id = "AhmedSSoliman/Llama2-CodeGen-PEFT-QLora"
+peft_model_id = "AhmedSSoliman/Llama2-CodeGen-PEFT-QLoRA"
 config = PeftConfig.from_pretrained(peft_model_id)
 model = AutoModelForCausalLM.from_pretrained(config.base_model_name_or_path, trust_remote_code=True, return_dict=True, load_in_4bit=True, device_map='auto')
 tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)