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

Added assertions

Browse Source
This commit is contained in:
Yuta Hayashibe
2023-10-26 13:34:36 +09:00
parent b2626bc7a9
commit 837a4dddb8
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
library/train_util.py
View File

@@ -148,6 +148,13 @@ class ImageInfo:
class BucketManager: class BucketManager:
def __init__(self, no_upscale, max_reso, min_size, max_size, reso_steps) -> None: def __init__(self, no_upscale, max_reso, min_size, max_size, reso_steps) -> None:
if max_size is not None:
if max_reso is not None:
assert max_size >= max_reso[0], "the max_size should be larger than the width of max_reso"
assert max_size >= max_reso[1], "the max_size should be larger than the height of max_reso"
if min_size is not None:
assert max_size >= min_size, "the max_size should be larger than the min_size"
self.no_upscale = no_upscale self.no_upscale = no_upscale
if max_reso is None: if max_reso is None:
self.max_reso = None self.max_reso = None