Alice - Personal Assistant Fine-Tuned
Este é um modelo baseado no GPT-2 fine-tunado para interações personalizadas para ser um IA pessoal de Leandro.
Uso
from transformers import AutoModelForCausalLM, AutoTokenizer
import torch
tokenizer = AutoTokenizer.from_pretrained("LPs02/Alice-model")
model = AutoModelForCausalLM.from_pretrained("LPs02/Alice-model")
inputs = tokenizer("Leandro: Oi Alice, tudo bem?\nAlice:", return_tensors="pt")
outputs = model.generate(**inputs, max_new_tokens=50)
print(tokenizer.decode(outputs[0]))
- Downloads last month
- 5