Llama-3.1-8B Mental Support
Fine-tuned on ShenLab/MentalChat16K dataset using LoRA/PEFT.
Usage
from transformers import AutoModelForCausalLM, AutoTokenizer
from peft import PeftModel
base = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-3.1-8B")
model = PeftModel.from_pretrained(base, "Rudraksh225/llama-3.1-8b-mental-support")
tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.1-8B")
- Downloads last month
- 42
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support
Model tree for Rudraksh225/llama-3.1-8b-mental-support
Base model
meta-llama/Llama-3.1-8B