Views
No views yet
fadodr/mental_health_therapyfadodr/mental_health_therapyInstruction: [Task or situation]
User: [User query or context]
Therapist: [Expected response]Instruction: <instruction>
User: <input>
Therapist: <output>| Parameter | Value |
|---|---|
| Method | QLoRA |
Rank (r) | 8 |
| LoRA Alpha | 16 |
| Dropout | 0.05 |
| Batch Size | 1 |
| Gradient Accumulation | 8 |
| Learning Rate | 2e-4 |
| Max Steps | 200 |
| Precision | FP16 |
| Save Steps | 25 |
1from transformers import AutoTokenizer, AutoModelForCausalLM
2import torch
3
4model_name = "your-username/mistral-mental-health-v0.1"
5
6tokenizer = AutoTokenizer.from_pretrained(model_name)
7model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto", torch_dtype=torch.float16)
8
9prompt = "I've been feeling very anxious about my studies. What can I do to calm myself down?"
10inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
11
12outputs = model.generate(**inputs, max_new_tokens=200, temperature=0.7, top_p=0.9)
13print(tokenizer.decode(outputs[0], skip_special_tokens=True))Nitin Tiwari. (2025). Mistral Mental Health Therapy v0.1. Hugging Face.
https://huggingface.co/your-username/mistral-mental-health-v0.1