mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-04-10 23:01:22 +00:00
fix: handle missing tokenizers in FluxTokenizeStrategy with fallbacks
Co-authored-by: aider (deepseek/deepseek-chat) <aider@aider.chat>
This commit is contained in:
@@ -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,
|
||||
|
||||
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user