github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Update test weights

Browse Source
This commit is contained in:
Jaret Burkett
2025-11-23 10:51:50 -07:00
parent 0b5f3475e2
commit dadbeda197
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -48,7 +48,7 @@ scheduler_config = {
MISTRAL_PATH = "mistralai/Mistral-Small-3.1-24B-Instruct-2503" MISTRAL_PATH = "mistralai/Mistral-Small-3.1-24B-Instruct-2503"
FLUX2_VAE_FILENAME = "flux2-vae.sft" 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) HF_TOKEN = os.getenv("HF_TOKEN", None)

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

@@ -471,7 +471,7 @@ export const modelArchs: ModelArch[] = [
group: 'image', group: 'image',
defaults: { defaults: {
// default updates when [selected, unselected] in the UI // 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': [true, false],
'config.process[0].model.quantize_te': [true, false], 'config.process[0].model.quantize_te': [true, false],
'config.process[0].model.low_vram': [true, false], 'config.process[0].model.low_vram': [true, false],