Views
No views yet
Pygmalion-6b, originally developed by the PygmalionAI team. It is designed for conversational AI and text-generation tasks, tailored to represent the persona of Peaches Sinclair, a charming and slightly clumsy Catgirl.Pygmalion-6b model was developed for high-quality conversational AI. This version incorporates a tailored persona for Peaches Sinclair, making it suitable for creative and engaging dialogues.Pygmalion-6b) and its training data. Users should carefully monitor outputs, especially for sensitive topics.1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_name = "WackyArt/Peaches-Pygmalion-6b"
4tokenizer = AutoTokenizer.from_pretrained(model_name)
5model = AutoModelForCausalLM.from_pretrained(model_name)
6
7inputs = tokenizer("Hello! How are you today?", return_tensors="pt")
8outputs = model.generate(**inputs)
9print(tokenizer.decode(outputs[0]))