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ürasyonu
  • adapter_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
Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support