BleachNick/SD3_UltraEdit_con_máscara

BleachNick

Texto a imagen

Modelo StableDiffusion3 entrenado con datos de UltraEdit para realizar ediciones de imagen basadas en máscara y en formas libres. Permite modificar imágenes usando un conjunto de instrucciones específicas.

Como usar

Para empezar a usar el modelo, sigue el siguiente código:
import torch
from diffusers import StableDiffusion3InstructPix2PixPipeline
from diffusers.utils import load_image
import requests
import PIL.Image

pipe = StableDiffusion3InstructPix2PixPipeline.from_pretrained("BleachNick/SD3_UltraEdit_w_mask", torch_dtype=torch.float16)
pipe = pipe.to("cuda")
prompt="¿Qué pasa si el caballo lleva un sombrero?"
img = load_image("input.png").resize((512, 512))
mask_img = load_image("mask_img.png").resize(img.size)
# Para edición en forma libre, usa una máscara en blanco
# mask_img = PIL.Image.new("RGB", img.size, (255, 255, 255))
image = pipe(
  prompt,
  image=img,
  mask_img=mask_img,
  negative_prompt="",
  num_inference_steps=50,
  image_guidance_scale=1.5,
  guidance_scale=7.5,
).images[0]
image.save("edited_image.png")
# muestra la imagen

Funcionalidades

Edición de imágenes basada en máscara
Edición de imágenes en forma libre
Integración con diffusers y PyTorch
Capacidad para trabajar con imágenes de entrada y máscaras

Casos de uso

Edición avanzada de imágenes con instrucciones específicas
Generación de imágenes modificadas para diseño gráfico
Pruebas en diseño visual con diferentes elementos gráficos