kohya-ss / kohya-ss/sd-scripts

Support schnell lora training

Open
#1,827 2 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

Hi @kohya-ss, I just find the current training script or code cannot be applied to FLUX schnell directly. According to @sdbds, it seems the shift problem? There are a lot of users prefer schnell, can you help to launch a schnell version for lora training ? that will be very helpful !!!!!!
I change the script to the following, but it still gets bad result.
```
--timestep_sampling uniform \
--discrete_flow_shift 1.0 \
--guidance_scale 0.0 \
```
![image](https://github.com/user-attachments/assets/f8abf9d0-0daf-40d9-a5c4-d6dfdf3b9ef1)

Contributor guide

No contributing guide indexed for this repository

Research direction

The issue does not name a file, entry point, or test. Start by locating the existing FLUX LoRA training path and the handling of --timestep_sampling, --discrete_flow_shift, and --guidance_scale; compare it with the FLUX schnell requirements described in the issue. Done means schnell training produces acceptable LoRA results and is covered by an appropriate validation or regression test.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
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.