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

Update train_util.py

Browse Source
This commit is contained in:
青龍聖者@bdsqlsz
2023-05-16 00:09:53 +08:00
committed by GitHub
parent 45f9cc9e0e
commit 7e5b6154d0
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
library/train_util.py
View File

@@ -3308,15 +3308,15 @@ def sample_images(
# prompts = f.readlines() # prompts = f.readlines()
if args.sample_prompts.endswith(".txt"): if args.sample_prompts.endswith(".txt"):
with open(args.sample_prompts, "r") as f: with open(args.sample_prompts, "r", encoding="utf-8") as f:
lines = f.readlines() lines = f.readlines()
prompts = [line.strip() for line in lines if len(line.strip()) > 0 and line[0] != "#"] prompts = [line.strip() for line in lines if len(line.strip()) > 0 and line[0] != "#"]
elif args.sample_prompts.endswith(".toml"): elif args.sample_prompts.endswith(".toml"):
with open(args.sample_prompts, "r") as f: with open(args.sample_prompts, "r", encoding="utf-8") 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"): elif args.sample_prompts.endswith(".json"):
with open(args.sample_prompts, "r") as f: with open(args.sample_prompts, "r", encoding="utf-8") as f:
prompts = json.load(f) prompts = json.load(f)
# schedulerを用意する # schedulerを用意する