nitrosocke/redshift-diffusion

nitrosocke
Texto a imagen

Modelo Stable Diffusion ajustado para generar imágenes con estética de render 3D de alta resolución inspirada en Redshift. Está entrenado para responder al token `redshift style`, que aplica un acabado visual de arte 3D/renderizado a personajes, vehículos, paisajes y escenas fantásticas.

Como usar

Instalación y uso con Diffusers:

pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline

# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained(
    "nitrosocke/redshift-diffusion",
    dtype=torch.bfloat16,
    device_map="cuda"
)

prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]

Ejemplo recomendado por la tarjeta del modelo usando el token de estilo:

from diffusers import StableDiffusionPipeline
import torch

model_id = "nitrosocke/redshift-diffusion"
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
pipe = pipe.to("cuda")

prompt = "redshift style magical princess with golden hair"
image = pipe(prompt).images[0]
image.save("./magical_princess.png")

Para obtener el efecto característico, incluye redshift style en el prompt. Ejemplos de prompts de la tarjeta: (redshift style) robert downey jr as ironman y redshift style Ford Mustang.

Funcionalidades

Generación de imágenes texto-a-imagen basada en Stable Diffusion.
Ajuste fino sobre obras 3D de alta resolución para obtener un estilo de render Redshift.
Uso del token `redshift style` en los prompts para activar el efecto estilístico principal.
Compatible con Diffusers mediante `DiffusionPipeline` o `StableDiffusionPipeline`.
Puede usarse como otros modelos Stable Diffusion y exportarse a ONNX, MPS o FLAX/JAX.
Licencia CreativeML OpenRAIL-M con uso comercial permitido bajo restricciones de uso responsable.

Casos de uso

Crear arte 3D estilizado a partir de texto.
Generar personajes con apariencia de render cinematográfico.
Crear vehículos y paisajes con estética Redshift.
Prototipar imágenes fantásticas, concept art y renders estilizados.
Usar un modelo Stable Diffusion especializado en pipelines locales con Diffusers.