RomanNum3ral
/
ai-toolkit
mirror of https://github.com/ostris/ai-toolkit.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update test weights

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
Show Stats 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"
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)

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

@ -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],