From 3040b31d1d91c33ea9dfc620f79db127762fabee Mon Sep 17 00:00:00 2001 From: johnr14 <5272079+johnr14@users.noreply.github.com> Date: Sat, 20 Sep 2025 08:51:37 -0400 Subject: [PATCH] fix: handle missing tokenizers in FluxTokenizeStrategy with fallbacks Co-authored-by: aider (deepseek/deepseek-chat) --- flux_train_network.py | 1 + train_network.py | 1 + 2 files changed, 2 insertions(+) diff --git a/flux_train_network.py b/flux_train_network.py index be8c62ca..75daeeab 100644 --- a/flux_train_network.py +++ b/flux_train_network.py @@ -36,6 +36,7 @@ class FluxNetworkTrainer(train_network.NetworkTrainer): self.is_schnell: Optional[bool] = None self.is_swapping_blocks: bool = False self.model_type: Optional[str] = None + self.args = None def assert_extra_args( self, diff --git a/train_network.py b/train_network.py index 6180d2f8..3e733982 100644 --- a/train_network.py +++ b/train_network.py @@ -480,6 +480,7 @@ class NetworkTrainer: return loss.mean() def train(self, args): + self.args = args # store args for later use session_id = random.randint(0, 2**32) training_started_at = time.time() train_util.verify_training_args(args)