Update README.md
Browse files
README.md
CHANGED
@@ -13,7 +13,7 @@ library_name: peft
|
|
13 |
language:
|
14 |
- en
|
15 |
datasets:
|
16 |
-
-
|
17 |
model_name: Meet Lora - Your Own Therapist
|
18 |
model_type: llama
|
19 |
inference: false
|
@@ -27,7 +27,7 @@ It leverages **Parameter-Efficient Fine-Tuning (PEFT)** using **LoRA (Low-Rank A
|
|
27 |
|
28 |
The model was fine-tuned on a **Tesla T4 GPU**, selectively training a small number of parameters to efficiently adapt the base model to the domain of mental health and therapy.
|
29 |
|
30 |
-
> 💡 Dataset used is available on [Hugging Face 🤗](https://huggingface.co/datasets/
|
31 |
|
32 |
---
|
33 |
|
@@ -55,7 +55,7 @@ base_model = AutoModelForCausalLM.from_pretrained(
|
|
55 |
tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-hf")
|
56 |
|
57 |
# Load the adapter
|
58 |
-
model = PeftModel.from_pretrained(base_model, "
|
59 |
|
60 |
# System prompt to guide behavior
|
61 |
system_prompt = (
|
|
|
13 |
language:
|
14 |
- en
|
15 |
datasets:
|
16 |
+
- kunalchamoli/mental_health_v1
|
17 |
model_name: Meet Lora - Your Own Therapist
|
18 |
model_type: llama
|
19 |
inference: false
|
|
|
27 |
|
28 |
The model was fine-tuned on a **Tesla T4 GPU**, selectively training a small number of parameters to efficiently adapt the base model to the domain of mental health and therapy.
|
29 |
|
30 |
+
> 💡 Dataset used is available on [Hugging Face 🤗](https://huggingface.co/datasets/kunalchamoli/mental_health_v1)
|
31 |
|
32 |
---
|
33 |
|
|
|
55 |
tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-hf")
|
56 |
|
57 |
# Load the adapter
|
58 |
+
model = PeftModel.from_pretrained(base_model, "aryan27/llama-therapy-lora")
|
59 |
|
60 |
# System prompt to guide behavior
|
61 |
system_prompt = (
|