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!