kohya-ss / kohya-ss/sd-scripts

Support Flux2

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

Description

Flux2 is pretty similar to the architecture for Flux1.

- Uses Mistral3.1 Small 24B for the text encoder
- New VAE for Flux2

They have a BFL reference based implementation of the transformer but haven't released the code for that. Because it's similar and they released the model, it should be available soon I'd imagine.

Contributor guide

No contributing guide indexed for this repository

Research direction

No files, tests, or entry points are identified. Start by reading the existing Flux1 support and the released Flux2 model materials, focusing on the Mistral3.1 Small 24B text encoder and new VAE; completion depends on the BFL transformer implementation becoming available and defining what full Flux2 support requires.

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.