Downloads · 30 days
0
Ozaii/zephyr-bae
zephyr-bae is a machine learning model from Ozaii. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
This is a fine-tuned version of Zephyr, optimized for romantic and friendly conversations.
Downloads · 30 days
0
Access
Public
Updated Aug 11, 2024
Repo size
336 MB
Likes
0
Public
Click a slice to open those files.
.safetensors336 MB · 97%
From the Hugging Face model README
This is a fine-tuned version of Zephyr, optimized for romantic and friendly conversations.
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!