From dadbeda1978deccee2f572ee34c90d49dac69675 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Sun, 23 Nov 2025 10:51:50 -0700 Subject: [PATCH] Update test weights --- extensions_built_in/diffusion_models/flux2/flux2_model.py | 2 +- ui/src/app/jobs/new/options.ts | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/extensions_built_in/diffusion_models/flux2/flux2_model.py b/extensions_built_in/diffusion_models/flux2/flux2_model.py index 11154e4d..ca48ebd9 100644 --- a/extensions_built_in/diffusion_models/flux2/flux2_model.py +++ b/extensions_built_in/diffusion_models/flux2/flux2_model.py @@ -48,7 +48,7 @@ scheduler_config = { MISTRAL_PATH = "mistralai/Mistral-Small-3.1-24B-Instruct-2503" FLUX2_VAE_FILENAME = "flux2-vae.sft" -FLUX2_TRANSFORMER_FILENAME = "flux-dev-dummy.sft" +FLUX2_TRANSFORMER_FILENAME = "flux2-final-dev.sft" 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 d328a2bc..4c730a5e 100644 --- a/ui/src/app/jobs/new/options.ts +++ b/ui/src/app/jobs/new/options.ts @@ -471,7 +471,7 @@ export const modelArchs: ModelArch[] = [ group: 'image', defaults: { // default updates when [selected, unselected] in the UI - 'config.process[0].model.name_or_path': ['ai-toolkit/f2-dummy', defaultNameOrPath], + 'config.process[0].model.name_or_path': ['ostris/f2', defaultNameOrPath], 'config.process[0].model.quantize': [true, false], 'config.process[0].model.quantize_te': [true, false], 'config.process[0].model.low_vram': [true, false],