Update README.md

README.md

@@ -1,10 +1,10 @@
----
-license: apache-2.0
-language:
-- en
-library_name: peft
-pipeline_tag: text-generation
----
+---
+license: apache-2.0
+language:
+- en
+library_name: peft
+pipeline_tag: text-generation
+---
 
 
 # Load the Model
@@ -14,7 +14,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
 
-base_model_path = "/storage/ukp/shared/shared_model_weights/models--llama-2-hf/7B"
+base_model_path = "meta-llama/Llama-2-7b-hf"
 model = AutoModelForCausalLM.from_pretrained(
             base_model_path,
             torch_dtype=torch.bfloat16,