trinart_stable_diffusion_v2
naclbit
Texto a imagen
trinart_stable_diffusion es un modelo de Diffusion Estable afinado con aproximadamente 40,000 imágenes de alta resolución con estilo anime/manga durante 8 épocas. Este es el mismo modelo que se ejecuta en el bot de Twitter @trinsama. La versión V2 utiliza dropouts, 10,000 imágenes adicionales y una nueva estrategia de etiquetado y se entrenó más tiempo para mejorar los resultados manteniendo la estética original.
Como usar
# Ejemplo de uso Text2Image
# !pip install diffusers==0.3.0
from diffusers import StableDiffusionPipeline
# usando el checkpoint de 60,000 pasos
pipe = StableDiffusionPipeline.from_pretrained('naclbit/trinart_stable_diffusion_v2', revision='diffusers-60k')
pipe.to('cuda')
image = pipe('Un dragón mágico volando frente al Himalaya en estilo manga').images[0]
image
# Ejemplo de uso Image2Image
# !pip install diffusers==0.3.0
from diffusers import StableDiffusionImg2ImgPipeline
import requests
from PIL import Image
from io import BytesIO
url = 'https://scitechdaily.com/images/Dog-Park.jpg'
response = requests.get(url)
init_image = Image.open(BytesIO(response.content)).convert('RGB')
init_image = init_image.resize((768, 512))
# usando el checkpoint de 115,000 pasos
pipe = StableDiffusionImg2ImgPipeline.from_pretrained('naclbit/trinart_stable_diffusion_v2', revision='diffusers-115k')
pipe.to('cuda')
images = pipe(prompt='Dibujo manga de Brad Pitt', init_image=init_image, strength=0.75, guidance_scale=7.5).images
images
Funcionalidades
- Modelo afinado con 40,000 imágenes de alta resolución de estilo anime/manga.
- Versión V2 incluye dropouts, 10,000 imágenes adicionales y una nueva estrategia de etiquetado.
- Portado a diffusers por ayan4m1.
- Checkpoints entrenados a 60,000, 95,000 y 115,000 pasos.
- Interfaz web de Gradio soportada.
Casos de uso
- Generación de imágenes en estilo manga o anime.
- Aplicaciones artísticas y creativas que requieran el estilo visual de manga/anime.
- Creación de contenidos gráficos para medios sociales inspirados en el estilo artístico del manga.