File size: 288 Bytes
028d637
 
 
 
 
 
1
2
3
4
5
6
7
from transformers import AutoModelForText2Image, AutoTokenizer

# Reemplaza "modelo-texto-imagen" con el nombre específico del modelo que deseas usar
model_name = "imagenes"
model = AutoModelForText2Image.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)