huggingface / huggingface/diffusers

Add aspect ratio bucketing to training scripts

Đang mở
#7,908 11 bình luận 3 reaction 0 người được giao Xem trên GitHub
stale
Ngôn ngữ chính
Python
Star
34.5k
Fork
7.3k
Merge trung bình
3 ngày 3 giờ
Pull request đã merge (30 ngày)
91

Mô tả

**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

Hướng dẫn đóng góp

Mở hướng dẫn đóng góp

Hướng nghiên cứu

Bắt đầu bằng cách xác định các script huấn luyện SDXL và phần mã chịu trách nhiệm áp đặt kích thước ảnh cố định và cắt ảnh. Xác định cách đối số tùy chọn --enable-bucketing phải chọn nhiều kích thước ảnh, sau đó kiểm tra rằng quá trình huấn luyện có thể giữ nguyên các tỷ lệ khung hình khác nhau mà không yêu cầu mọi ảnh đều bị cắt thành 1024x1024.

Do mô hình lập chỉ mục viết ra từ nội dung của issue.

Đánh giá

Công nghệ
python, pytorch
Lĩnh vực
machine-learning
Loại issue
Tính năng
Độ khó
5/5
Thời gian dự kiến
Hơn một tuần
Mức độ hoạt động
Đình trệ
Độ rõ ràng
Cần làm rõ
Mức phù hợp với người mới
35/100

Nhận issue mới trong hộp thư của bạn

Bản tóm tắt ngắn những issue GitHub phù hợp với người mới.