lora-ropa-deportiva-sd21

LoRA Model - Johan Sport Ropa Deportiva

Modelo LoRA entrenado con 1,403 imágenes de prendas deportivas de la microempresa Johan Sport para la generación de diseños personalizados de ropa deportiva.

Detalles del Entrenamiento

• Modelo base: Stable Diffusion 2.1
• Técnica: LoRA (Low-Rank Adaptation)
• Dataset: 1,403 imágenes de prendas deportivas
• Resolución: 512x512 px
• Pasos de entrenamiento: 100 steps

Uso en Python

from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler import torch

Cargar modelo base pipe = StableDiffusionPipeline.from_pretrained( "stabilityai/stable-diffusion-2-1", torch_dtype=torch.float16, use_auth_token="YOUR_HF_TOKEN" )

Cargar adaptador LoRA pipe.unet.load_attn_procs( "AlexanderCraf/stable-diffusion-johanSport-lora", weight_name="lora_ropa_test.safetensors", use_auth_token="YOUR_HF_TOKEN" )

pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config) pipe.to("cuda")

Generar imagen prompt = "camiseta deportiva roja con logo en el pecho, diseño moderno" image = pipe(prompt, num_inference_steps=25).images image.save("resultado.png")

Aplicación

Este modelo fue desarrollado como parte del Trabajo de Integración Curricular para el aplicativo web de comercialización de ropa deportiva de la microempresa Johan Sport, Tulcán - Ecuador.