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

feat: added json support as well

Browse Source
This commit is contained in:
Linaqruf
2023-05-14 19:49:54 +07:00
parent 774c4059fb
commit 8ab5c8cb28
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
library/train_util.py
View File

@@ -3303,6 +3303,9 @@ def sample_images(
with open(args.sample_prompts, 'r') as f: with open(args.sample_prompts, 'r') as f:
data = toml.load(f) data = toml.load(f)
prompts = [dict(**data['prompt'], **subset) for subset in data['prompt']['subset']] 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を用意する # schedulerを用意する
sched_init_args = {} sched_init_args = {}