File size: 851 Bytes
f69b0ec ff91439 f69b0ec b77ba80 f69b0ec |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
---
tags:
- psychology
- chatbot
- llama
- therapy
- mental health
license: llama3.1
language:
- en
base_model:
- meta-llama/Llama-3.1-8B
---
# WellMinded Therapy Engine (8B)
This is a fine-tuned version of the LLaMA 3.1 8B model, optimized for psychologist-like conversations. The model is quantized to 4-bit precision (Q4_0) for efficient inference.
## Usage
You can load and use this model with the `transformers` library:
```python
from transformers import AutoModelForCausalLM, AutoTokenizer
model = AutoModelForCausalLM.from_pretrained("PixelPanda1/WellMinded_Therapy_Engine")
tokenizer = AutoTokenizer.from_pretrained("PixelPanda1/WellMinded_Therapy_Engine")
input_text = "Hi, I'm feeling really stressed lately."
inputs = tokenizer(input_text, return_tensors="pt")
outputs = model.generate(**inputs)
print(tokenizer.decode(outputs[0])) |