aryan27 commited on
Commit
dfc41fe
·
verified ·
1 Parent(s): 0a7b5ae

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -13,7 +13,7 @@ library_name: peft
13
  language:
14
  - en
15
  datasets:
16
- - your-username/therapy-dataset
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/your-username/therapy-dataset)
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, "your-username/meet-lora-therapist")
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 = (