ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-01-26 16:39:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add FLUX.2 official weights

Browse Source
This commit is contained in:
Jaret Burkett
2025-11-25 08:52:19 -07:00
parent dadbeda197
commit 01cf480233
3 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
extensions_built_in/diffusion_models/flux2/flux2_model.py
View File

@@ -47,8 +47,8 @@ scheduler_config = {
}
MISTRAL_PATH = "mistralai/Mistral-Small-3.1-24B-Instruct-2503"
FLUX2_VAE_FILENAME = "flux2-vae.sft"
FLUX2_TRANSFORMER_FILENAME = "flux2-final-dev.sft"
FLUX2_VAE_FILENAME = "ae.safetensors"
FLUX2_TRANSFORMER_FILENAME = "flux2-dev.safetensors"
HF_TOKEN = os.getenv("HF_TOKEN", None)

4
ui/src/app/jobs/new/options.ts
View File

@@ -467,11 +467,11 @@ export const modelArchs: ModelArch[] = [
},
{
name: 'flux2',
label: 'FLUX.2(DUMMY)',
label: 'FLUX.2',
group: 'image',
defaults: {
// default updates when [selected, unselected] in the UI
'config.process[0].model.name_or_path': ['ostris/f2', defaultNameOrPath],
'config.process[0].model.name_or_path': ['black-forest-labs/FLUX.2-dev', defaultNameOrPath],
'config.process[0].model.quantize': [true, false],
'config.process[0].model.quantize_te': [true, false],
'config.process[0].model.low_vram': [true, false],

2
version.py
View File

@@ -1 +1 @@
VERSION = "0.7.4"
VERSION = "0.7.5"