| 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 | |
| ```python | |
| 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! | |