YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
Postpartum Emotional Support Model (LoRA)
Model Açıklaması
Bu model, hamilelik ve doğum sonrası dönemde annelere duygusal destek sağlamak için fine-tune edilmiştir. StableLM-2-Zephyr-1.6B base model'i üzerine LoRA ile eğitilmiştir.
Kullanım
LoRA Adaptörü ile Kullanım
from transformers import AutoTokenizer, AutoModelForCausalLM
from peft import PeftModel
# Base model'i yükle
base_model = AutoModelForCausalLM.from_pretrained(
"stabilityai/stablelm-2-zephyr-1_6b",
torch_dtype=torch.float16,
device_map="auto"
)
# Tokenizer'ı yükle
tokenizer = AutoTokenizer.from_pretrained("stabilityai/stablelm-2-zephyr-1_6b")
# LoRA adaptörünü yükle
model = PeftModel.from_pretrained(base_model, "rima0034/postpartum_emotional_support")
# Sohbet için hazırla
messages = [
{"role": "system", "content": "You are an empathetic emotional support assistant for mothers."},
{"role": "user", "content": "I'm feeling overwhelmed with motherhood"}
]
# Yanıt üret
formatted_prompt = tokenizer.apply_chat_template(messages, tokenize=False)
inputs = tokenizer(formatted_prompt, return_tensors="pt")
outputs = model.generate(**inputs, max_new_tokens=256)
response = tokenizer.decode(outputs[0], skip_special_tokens=True)
Model Detayları
- Base Model: stabilityai/stablelm-2-zephyr-1_6b
- Fine-tuning: LoRA (Low-Rank Adaptation)
- Adaptör Boyutu: ~12MB
- Eğitim Verisi: Postpartum duygusal destek konuşmaları
- Dil: İngilizce
Dosyalar
adapter_config.json- LoRA konfigürasyonuadapter_model.safetensors- LoRA adaptör ağırlıkları
Lisans
Bu model eğitim amaçlı kullanım için geliştirilmiştir. Tıbbi tavsiye yerine geçmez.
Katkıda Bulunanlar
- Rima0034 - Model geliştirme ve eğitim
- Hugging Face - Altyapı desteği
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support