Hugging Face's logo

Muhammad7865253
/
gemma-2-2b-it-fine-tune-model-mental-health

Text Generation
Transformers
TensorBoard
Safetensors
gemma
conversational
text-generation-inference
Inference Endpoints
4-bit precision
bitsandbytes
Model card Files Metrics Community

You need to agree to share your contact information to access this model

This repository is publicly accessible, but you have to accept the conditions to access its files and content.

Log in or Sign Up to review the conditions and access this model content.

Model Name: Gemma-2-2B-IT

This is a fine-tuned model based on Gemma-2-2B-IT, optimized for text generation tasks. It is compatible with Hugging Face's transformers library.

Fine-Tuning Details

  • Dataset: Mental Health / Therapist Dataset
  • Method: LoRA for low-resource adaptation

Usage Example 

from transformers import AutoModelForCausalLM, AutoTokenizer

model = AutoModelForCausalLM.from_pretrained("YourModelRepoName")
tokenizer = AutoTokenizer.from_pretrained("YourModelRepoName")

input_text = "What is the meaning of life?"
inputs = tokenizer(input_text, return_tensors="pt")
outputs = model.generate(**inputs)
print(tokenizer.decode(outputs[0]))
Downloads last month
34
Safetensors
Model size
1.55B params
Tensor type
F32
·
U8
·
Inference Examples
Text Generation
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Dataset used to train Muhammad7865253/gemma-2-2b-it-fine-tune-model-mental-health