Commit
·
7c60e5e
1
Parent(s):
225d66b
Update README.md
Browse files
README.md
CHANGED
|
@@ -108,7 +108,7 @@ MODEL_NAME = "Sandiago21/llama-13b-hf-prompt-answering"
|
|
| 108 |
config = PeftConfig.from_pretrained(MODEL_NAME)
|
| 109 |
|
| 110 |
# Setting the path to look at your repo directory, assuming that you are at that directory when running this script
|
| 111 |
-
config.base_model_name_or_path = "decapoda-research/llama-
|
| 112 |
|
| 113 |
model = LlamaForCausalLM.from_pretrained(
|
| 114 |
config.base_model_name_or_path,
|
|
|
|
| 108 |
config = PeftConfig.from_pretrained(MODEL_NAME)
|
| 109 |
|
| 110 |
# Setting the path to look at your repo directory, assuming that you are at that directory when running this script
|
| 111 |
+
config.base_model_name_or_path = "decapoda-research/llama-13b-hf/"
|
| 112 |
|
| 113 |
model = LlamaForCausalLM.from_pretrained(
|
| 114 |
config.base_model_name_or_path,
|