kohya-ss / kohya-ss/sd-scripts

Possible Enhancements/optimizations. - Is quantization possible for kohya? ai toolkit has it. Serious speed boost for 1024 training, and bucketing multi resolutions.

Open
#1,503 3 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
7.2k
Forks
1.2k
Avg merge
11m
Merged PRs (30d)
2

Description

1. Is it possible to add quantization? with ai toolkit i get 22.1gb vram doing 1024 training, and 1400 steps done in 45 mins
in kohya 1024 training, takes me nearly 2 hours with the same amount.

2. Multiresolution training with bucketing, it buckets all my images into 512, 768, and 1024.

overall I'm getting quite good results.

Contributor guide

No contributing guide indexed for this repository

Research direction

The issue names no files, tests, or entry points. Start by reviewing the existing training and resolution-bucketing flow, then compare it with the ai toolkit's quantization and bucketing behavior. Done means the requested quantization and multiresolution bucketing are supported with improved training speed or VRAM use.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning, performance
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.