rojasdiego
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -17,7 +17,7 @@ This model is QLORA adapater for Meta's LLaMa 3.1 8B model, trained to answer qu
|
|
17 |
|
18 |
## Dataset
|
19 |
|
20 |
-
Fine-tuned on a single epoch of [Apple MLX QA](https://huggingface.co/datasets/
|
21 |
|
22 |
## Installation
|
23 |
|
@@ -45,7 +45,7 @@ tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.1-8B-Instruct")
|
|
45 |
# Load the fine-tuned model using LORA
|
46 |
model = PeftModel.from_pretrained(
|
47 |
model,
|
48 |
-
"
|
49 |
).to("cuda")
|
50 |
|
51 |
# Define input using a chat template with a system prompt and user query
|
|
|
17 |
|
18 |
## Dataset
|
19 |
|
20 |
+
Fine-tuned on a single epoch of [Apple MLX QA](https://huggingface.co/datasets/koyeb/Apple-MLX-QA).
|
21 |
|
22 |
## Installation
|
23 |
|
|
|
45 |
# Load the fine-tuned model using LORA
|
46 |
model = PeftModel.from_pretrained(
|
47 |
model,
|
48 |
+
"koyeb/Meta-Llama-3.1-8B-Instruct-Apple-MLX",
|
49 |
).to("cuda")
|
50 |
|
51 |
# Define input using a chat template with a system prompt and user query
|