README.md

metadata

language: en
tags:
  - zephyr
  - conversational
license: mit

Zephyr AI Boyfriend

This is a fine-tuned version of Zephyr, optimized for romantic and friendly conversations.

Usage

from transformers import AutoModelForCausalLM, AutoTokenizer

model = AutoModelForCausalLM.from_pretrained("Ozzai/zephyr-bae")
tokenizer = AutoTokenizer.from_pretrained("Ozzai/zephyr-bae")

# Example usage
input_text = "Hey Zephyr, what's on your mind?"
inputs = tokenizer(input_text, return_tensors="pt")
outputs = model.generate(**inputs, max_new_tokens=100)
response = tokenizer.decode(outputs[0])
print(response)

Enjoy chatting with Zephyr!