ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-03-13 06:29:48 +00:00
Code Issues 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"
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)