ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-28 18:21:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add FLUX.2 official weights

Browse Source
This commit is contained in:
Jaret Burkett
2025-11-25 08:52:19 -07:00
parent dadbeda197
commit 01cf480233
3 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -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)