FLUX: save train config (#1049)

2025-09-05 00:04:38 +08:00 · 2024-11-09 09:15:19 +08:00
parent 657b4cc0aa
commit 1e07660184
4 changed files with 35 additions and 6 deletions
--- a/flux/flux/init.py
+++ b/flux/flux/init.py
@@ -12,4 +12,5 @@ from .utils import (
    load_flow_model,
    load_t5,
    load_t5_tokenizer,
+    save_config,
 )