VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-06 21:52:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix AttributeError: 'FluxNetworkTrainer' object has no attribute 'sample_prompts_te_outputs'

Browse Source 
Move "self.sample_prompts_te_outputs = None" from Line 150 to Line 26.
This commit is contained in:
DukeG
2024-08-16 11:00:08 +08:00
parent 35b6cb0cd1
commit 08ef886bfe
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
flux_train_network.py
View File

@@ -23,6 +23,7 @@ logger = logging.getLogger(__name__)
class FluxNetworkTrainer(train_network.NetworkTrainer):
def __init__(self):
super().__init__()
self.sample_prompts_te_outputs = None
def assert_extra_args(self, args, train_dataset_group):
super().assert_extra_args(args, train_dataset_group)
@@ -147,7 +148,6 @@ class FluxNetworkTrainer(train_network.NetworkTrainer):
dataset.new_cache_text_encoder_outputs(text_encoders, accelerator.is_main_process)
# cache sample prompts
self.sample_prompts_te_outputs = None
if args.sample_prompts is not None:
logger.info(f"cache Text Encoder outputs for sample prompt: {args.sample_prompts}")