This commit is contained in:
wailovet
2024-06-19 01:52:49 +08:00
parent 65135d70ae
commit 9c20d44949
+1
View File
@@ -236,6 +236,7 @@ import logging
def generate_toml_config(output_path, enable_bucket=True, resolution=512, batch_size=1, image_dir=None, caption_extension=".caption", num_repeats=10, ):
check_install()
import toml
config = {
'general': {