kohya-ss / kohya-ss/sd-scripts

"mixed_precision fp16" not working for Flux

Open
#1,707 4 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

Why doesn't mixed_precision fp16 work for Flux?
In SDXL it was possible to train in fp16 and the result was quite normal (probably, didn't compare).

If specify mixed_precision fp16 for Flux, the training goes normally and the training speed is **5 times faster** than with bf16, but as a result LoRa turns out to be "empty", it does not make any changes when connected.

By the way, how right or wrong is it to train in FP16? Do the gains in speed and resources justify the losses?

Contributor guide

No contributing guide indexed for this repository

Research direction

Start by reproducing Flux training with mixed_precision set to fp16, then compare the resulting LoRA with a bf16 run and the reported SDXL behavior. Done means identifying why the fp16 LoRA is empty or ineffective and documenting or correcting the behavior, with a reproducible comparison.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.