pixart-900m-1024-ft-v0.7-stage2

Importa y usa el modelo de la siguiente manera en Python:

import torch
from diffusers import DiffusionPipeline

model_id = 'pixart-900m-1024-ft-v0.7-stage2'
pipeline = DiffusionPipeline.from_pretrained(model_id)

prompt = "un personaje de anime lindo llamado toast, sosteniendo un cartel que dice SOON"
prompt_negativo = "borroso, recortado, feo"

pipeline.to('cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu')
imagen = pipeline(
  prompt=prompt,
  negative_prompt='borroso, recortado, feo',
  num_inference_steps=30,
  generator=torch.Generator(device='cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu').manual_seed(1641421826),
  width=1152,
  height=768,
  guidance_scale=4.0,
  guidance_rescale=0.7,
).images[0]
imagen.save("output.png", format="PNG")