pony-diffusion - >nohooves

AstraliteHeart
Texto a imagen

Pony Diffusion V4 está en vivo! pony-diffusion es un modelo latente de difusión de texto a imagen que ha sido condicionado en imágenes SFW-ish de ponis de alta calidad a través de un ajuste fino. Con un agradecimiento especial a Waifu-Diffusion por proporcionar la experiencia en ajuste fino y a Novel AI por proporcionar el cómputo necesario.

Como usar

import torch
from torch import autocast
from diffusers import StableDiffusionPipeline, DDIMScheduler
model_id = "AstraliteHeart/pony-diffusion"
device = "cuda"
pipe = StableDiffusionPipeline.from_pretrained(
model_id,
torch_dtype=torch.float16,
revision="fp16",
scheduler=DDIMScheduler(
beta_start=0.00085,
beta_end=0.012,
beta_schedule="scaled_linear",
clip_sample=False,
set_alpha_to_one=False,
),
)
pipe = pipe.to(device)
prompt = "pinkie pie anthro portrait wedding dress veil intricate highly detailed digital painting artstation concept art smooth sharp focus illustration Unreal Engine 5 8K"
with autocast("cuda"):
  image = pipe(prompt, guidance_scale=7.5)["sample"][0]  

image.save("cute_poner.png")

Funcionalidades

Modelo de difusión latente de texto a imagen
Condicionado en imágenes SFW-ish de ponis de alta calidad
Afinado con una tasa de aprendizaje de 5.0e-6 durante 4 épocas
Entrenamiento en aproximadamente 80k pares de texto-imagen de ponis

Casos de uso

Propósitos de entretenimiento
Asistente de arte generativo

Recibe las últimas noticias y actualizaciones sobre el mundo de IA directamente en tu bandeja de entrada.