huggingface / huggingface/diffusers

Add aspect ratio bucketing to training scripts

Abierto
#7,908 11 comentarios 3 reacciones 0 asignados Ver en GitHub
stale
Lenguaje dominante
Python
Estrellas
34.5k
Forks
7.3k
Merge medio
3 d 3 h
PR fusionados (30 d)
91

Descripción

**Is your feature request related to a problem? Please describe.**
When fine tuning SDXL, images are required to be a fixed size (1024x1024) which involves a lot of cropping that both takes time/resources, and often causes important parts of the image to get cropped out, which lowers model quality.

**Describe the solution you'd like.**
The ideal solution would be a simple option for user to enable aspect ratio bucketing (e.g. a command argument `--enable-bucketing`) that will let them train with multiple image sizes

Guía de contribución

Abrir la guía de contribución

Línea de trabajo

Start by locating the SDXL training scripts and the code responsible for enforcing fixed image sizes and cropping. Define how the optional --enable-bucketing argument should select multiple image sizes, then verify that training can preserve varied aspect ratios without requiring every image to be cropped to 1024x1024.

Escrito por el modelo de indexación a partir del texto del issue.

Evaluación

Stack tecnológico
python, pytorch
Área
machine-learning
Tipo de issue
Nueva funcionalidad
Dificultad
5/5
Tiempo estimado
Más de una semana
Estado de actividad
Estancado
Claridad
Necesita aclaración
Aptitud para principiantes
35/100

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.