Update README.md
Browse files
README.md
CHANGED
|
@@ -1,20 +1,20 @@
|
|
| 1 |
---
|
| 2 |
-
license: mit
|
| 3 |
-
datasets:
|
| 4 |
-
- Lumiiree/therapod-dpo
|
| 5 |
-
language:
|
| 6 |
-
- en
|
| 7 |
-
base_model:
|
| 8 |
-
- meta-llama/Llama-3.2-3B-Instruct
|
| 9 |
tags:
|
| 10 |
- llama
|
| 11 |
-
-
|
| 12 |
-
- therapy
|
| 13 |
-
- lora
|
| 14 |
-
- qlora
|
| 15 |
-
- casual-lm
|
| 16 |
- vllm
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 17 |
---
|
|
|
|
| 18 |
# 🧠 CBT-Copilot: LLaMA 3.2B Fine-Tuned for Cognitive Therapy
|
| 19 |
|
| 20 |
Welcome to **CBT-Copilot**, an open-source LLM fine-tuned on therapy-aligned dialogues using the [Lumiiree/therapod-dpo](https://huggingface.co/datasets/Lumiiree/therapod-dpo) dataset. This model is designed to act as a **compassionate and supportive AI assistant**, trained in the tone of cognitive behavioral therapy (CBT), and suitable for mental health support applications.
|
|
@@ -104,6 +104,40 @@ print(response[0]['generated_text'])
|
|
| 104 |
|
| 105 |
---
|
| 106 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 107 |
## 🧾 License
|
| 108 |
|
| 109 |
This project is open-sourced for educational and research purposes under the **MIT License**.
|
|
@@ -125,4 +159,4 @@ This project is open-sourced for educational and research purposes under the **M
|
|
| 125 |
|
| 126 |
---
|
| 127 |
|
| 128 |
-
*Crafted with care by Thillai Chithambaram for the future of compassionate AI.*
|
|
|
|
| 1 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
tags:
|
| 3 |
- llama
|
| 4 |
+
- llama-3
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
- vllm
|
| 6 |
+
- causal-lm
|
| 7 |
+
- cognitive-therapy
|
| 8 |
+
- mental-health
|
| 9 |
+
- conversational
|
| 10 |
+
- peft
|
| 11 |
+
- lora
|
| 12 |
+
license: mit
|
| 13 |
+
datasets:
|
| 14 |
+
- Lumiiree/therapod-dpo
|
| 15 |
+
inference: true
|
| 16 |
---
|
| 17 |
+
|
| 18 |
# 🧠 CBT-Copilot: LLaMA 3.2B Fine-Tuned for Cognitive Therapy
|
| 19 |
|
| 20 |
Welcome to **CBT-Copilot**, an open-source LLM fine-tuned on therapy-aligned dialogues using the [Lumiiree/therapod-dpo](https://huggingface.co/datasets/Lumiiree/therapod-dpo) dataset. This model is designed to act as a **compassionate and supportive AI assistant**, trained in the tone of cognitive behavioral therapy (CBT), and suitable for mental health support applications.
|
|
|
|
| 104 |
|
| 105 |
---
|
| 106 |
|
| 107 |
+
## ⚡ Inference with vLLM
|
| 108 |
+
|
| 109 |
+
This model is **compatible with [vLLM](https://github.com/vllm-project/vllm)** — a fast and memory-efficient inference engine for LLMs.
|
| 110 |
+
|
| 111 |
+
### 🚀 Quick Start with vLLM
|
| 112 |
+
|
| 113 |
+
```bash
|
| 114 |
+
pip install "vllm[serve]"
|
| 115 |
+
|
| 116 |
+
python -m vllm.entrypoints.api_server \
|
| 117 |
+
--model thillaic/CBT-Copilot
|
| 118 |
+
```
|
| 119 |
+
|
| 120 |
+
### OpenAI-Compatible API Example
|
| 121 |
+
|
| 122 |
+
```python
|
| 123 |
+
import openai
|
| 124 |
+
|
| 125 |
+
openai.api_base = "http://localhost:8000/v1"
|
| 126 |
+
openai.api_key = "EMPTY" # not needed for local use
|
| 127 |
+
|
| 128 |
+
response = openai.ChatCompletion.create(
|
| 129 |
+
model="thillaic/CBT-Copilot",
|
| 130 |
+
messages=[
|
| 131 |
+
{"role": "system", "content": "You are a compassionate CBT therapist."},
|
| 132 |
+
{"role": "user", "content": "I'm feeling anxious lately. What should I do?"}
|
| 133 |
+
]
|
| 134 |
+
)
|
| 135 |
+
|
| 136 |
+
print(response.choices[0].message["content"])
|
| 137 |
+
```
|
| 138 |
+
|
| 139 |
+
---
|
| 140 |
+
|
| 141 |
## 🧾 License
|
| 142 |
|
| 143 |
This project is open-sourced for educational and research purposes under the **MIT License**.
|
|
|
|
| 159 |
|
| 160 |
---
|
| 161 |
|
| 162 |
+
*Crafted with care by Thillai Chithambaram for the future of compassionate AI.*
|