Update README.md

README.md

@@ -11,7 +11,13 @@ pipeline_tag: text-generation
 base_model: meta-llama/Llama-2-7b-hf
 ---
 
-# llama-2-7b-instruct-v0.1 🦙
+<div align="center">
+
+<img src="./assets/llama.png" width="150px">
+
+</div>
+
+# Llama-2-7B-Instruct-v0.1
 
 This instruction model was built via parameter-efficient QLoRA finetuning of [Llama-2-7b](https://huggingface.co/meta-llama/Llama-2-7b-hf) on the first 5k rows of [ehartford/dolphin](https://huggingface.co/datasets/ehartford/dolphin) and the first 5k rows of [garage-bAInd/Open-Platypus](https://huggingface.co/datasets/garage-bAInd/Open-Platypus). Finetuning was executed on 1x A100 (40 GB SXM) for roughly 2 hours on the [Lambda Labs](https://cloud.lambdalabs.com/instances) platform.
 
@@ -72,7 +78,7 @@ notebook_login()
 ```
 
 ```python
-peft_model_id = "dfurman/llama-2-7b-instruct-v0.1"
+peft_model_id = "dfurman/Llama-2-7B-Instruct-v0.1"
 config = PeftConfig.from_pretrained(peft_model_id)
 
 bnb_config = BitsAndBytesConfig(