VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-08 22:35:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix warning messages are shown every step

Browse Source
This commit is contained in:
Kohya S
2023-06-26 20:37:14 +09:00
parent 747af145ed
commit 56ca5dfa15
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
sdxl_train_network.py
View File

@@ -8,6 +8,7 @@ class SdxlNetworkTrainer(train_network.NetworkTrainer):
def __init__(self): def __init__(self):
super().__init__() super().__init__()
self.vae_scale_factor = sdxl_model_util.VAE_SCALE_FACTOR self.vae_scale_factor = sdxl_model_util.VAE_SCALE_FACTOR
self.sampling_warning_showed = False
def assert_extra_args(self, args, train_dataset_group): def assert_extra_args(self, args, train_dataset_group):
super().assert_extra_args(args) super().assert_extra_args(args)
@@ -153,7 +154,9 @@ class SdxlNetworkTrainer(train_network.NetworkTrainer):
return noise_pred return noise_pred
def sample_images(self, accelerator, args, epoch, global_step, device, vae, tokenizer, text_encoder, unet): def sample_images(self, accelerator, args, epoch, global_step, device, vae, tokenizer, text_encoder, unet):
print("sample_images is not implemented") if not self.sampling_warning_showed:
print("sample_images is not implemented")
self.sampling_warning_showed = True
def setup_parser() -> argparse.ArgumentParser: def setup_parser() -> argparse.ArgumentParser: