metadata
license: mit
tags:
- text-generation-inference
pipeline_tag: text-generation
language:
- pt
base_model:
- openai-community/gpt2-large
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]))