huggingface / huggingface/diffusers
Add aspect ratio bucketing to training scripts
- 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
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