MindMate-AI: Mental Health Assistant Fine-Tuned on Llama-2-7b
π Features
- π§ CBT-focused response generation
- π 4-bit quantization with QLoRA
- π‘ Context-aware mental health dialogues
- π Optimized for therapeutic conversations
- π€ HF Transformers/trl/peft integration
π Requirements
Hardware
- NVIDIA GPU (16GB+ VRAM recommended)
- CUDA 11.7+
- Python 3.9+
Accounts
- Hugging Face account
- Access to Meta's Llama-2 models
- Kaggle account (for dataset access)
βοΈ Installation
# Base dependencies
pip install -q accelerate==0.21.0 peft==0.4.0 bitsandbytes==0.40.2 transformers==4.31.0 trl==0.4.7
# Additional requirements
pip install -q datasets
pip install -q tensorboard
# Environment setup
export HF_TOKEN="your_huggingface_token"
export KAGGLE_CONFIG_DIR="/path/to/kaggle.json"
## β οΈ Important Legal Disclaimers πΌ
**This is a non-commercial academic project. Not approved for medical use.**
```python
# Required in any interface code
print("DISCLAIMER: Academic prototype - outputs may be inaccurate. Not for real-world use.")
This project uses Meta's Llama-2-7b model under the [LLAMA 2 Community License](https://ai.meta.com/resources/models-and-libraries/llama-downloads/).
**You MUST:**
1. Maintain the original [Llama-2 LICENSE](https://github.com/facebookresearch/llama/blob/main/LICENSE) file
2. Keep this notice visible in all distributions
3. Not use for prohibited activities (see Meta's Acceptable Use Policy)
**Copyrights:**
- Base Model: Copyright Β© Meta Platforms, Inc.
## π Intended Use
- Experimental mental health conversation support only
- Strictly for academic research (psychology/NLP education)
## β οΈ Limitations & Risks
- May generate inaccurate or harmful mental health suggestions
- No clinical validity - do not use for diagnosis/therapy
- Potential bias in training data (document your dataset source)
## π‘οΈ Ethical Considerations
- Contains [Meta's Llama-2 safeguards](https://ai.meta.com/llama/use-policy/)
- Additional mental health-specific safety mitigations: [describe yours]
- Downloads last month
- 10
Inference Providers
NEW
This model is not currently available via any of the supported third-party Inference Providers, and
the model is not deployed on the HF Inference API.
Model tree for saarib2405/MindMate-AI
Base model
meta-llama/Llama-2-7b-chat-hf