From 01cf480233438578ecc8095b0f1d63c0a89594de Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Tue, 25 Nov 2025 08:52:19 -0700 Subject: [PATCH] Add FLUX.2 official weights --- extensions_built_in/diffusion_models/flux2/flux2_model.py | 4 ++-- ui/src/app/jobs/new/options.ts | 4 ++-- version.py | 2 +- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/extensions_built_in/diffusion_models/flux2/flux2_model.py b/extensions_built_in/diffusion_models/flux2/flux2_model.py index ca48ebd9..6467ead0 100644 --- a/extensions_built_in/diffusion_models/flux2/flux2_model.py +++ b/extensions_built_in/diffusion_models/flux2/flux2_model.py @@ -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) diff --git a/ui/src/app/jobs/new/options.ts b/ui/src/app/jobs/new/options.ts index 4c730a5e..5c7fe1f0 100644 --- a/ui/src/app/jobs/new/options.ts +++ b/ui/src/app/jobs/new/options.ts @@ -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], diff --git a/version.py b/version.py index 9b8720f5..c6746c8c 100644 --- a/version.py +++ b/version.py @@ -1 +1 @@ -VERSION = "0.7.4" \ No newline at end of file +VERSION = "0.7.5" \ No newline at end of file