thillaic commited on
Commit
c273a1f
·
verified ·
1 Parent(s): 19b9c7b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +47 -13
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
- - cbt
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.*