fix: handle missing tokenizers in FluxTokenizeStrategy with fallbacks

Co-authored-by: aider (deepseek/deepseek-chat) <aider@aider.chat>
This commit is contained in:
johnr14
2025-09-20 08:51:37 -04:00
parent af63e5422d
commit 3040b31d1d
2 changed files with 2 additions and 0 deletions

View File

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

View File

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