From 8ab5c8cb28c8efbe97c76b8abdb5a97779e5ac84 Mon Sep 17 00:00:00 2001 From: Linaqruf Date: Sun, 14 May 2023 19:49:54 +0700 Subject: [PATCH] feat: added json support as well --- library/train_util.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/library/train_util.py b/library/train_util.py index 438c397b..039f58b5 100644 --- a/library/train_util.py +++ b/library/train_util.py @@ -3303,6 +3303,9 @@ def sample_images( with open(args.sample_prompts, 'r') as f: data = toml.load(f) prompts = [dict(**data['prompt'], **subset) for subset in data['prompt']['subset']] + elif args.sample_prompts.endswith('.json'): + with open(args.sample_prompts, 'r') as f: + prompts = json.load(f) # schedulerを用意する sched_init_args = {}