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

fix to work

Browse Source
This commit is contained in:
Kohya S
2023-06-28 13:09:02 +09:00
parent 71a6d49d06
commit 8521ab7990
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
sdxl_minimal_inference.py
View File

@@ -169,7 +169,7 @@ if __name__ == "__main__":
beta_schedule=SCHEDLER_SCHEDULE, beta_schedule=SCHEDLER_SCHEDULE,
) )
def generate_image(text, negative_text, seed=None): def generate_image(prompt, negative_prompt, seed=None):
# 将来的にサイズ情報も変えられるようにする / Make it possible to change the size information in the future # 将来的にサイズ情報も変えられるようにする / Make it possible to change the size information in the future
# prepare embedding # prepare embedding
with torch.no_grad(): with torch.no_grad():
@@ -295,7 +295,7 @@ if __name__ == "__main__":
img.save(os.path.join(args.output_dir, f"image_{timestamp}_{i:03d}.png")) img.save(os.path.join(args.output_dir, f"image_{timestamp}_{i:03d}.png"))
if not args.interactive: if not args.interactive:
generate_image(args.prompt, args.negative_prompt, args.seed) generate_image(args.prompt, args.negative_prompt, seed)
else: else:
# loop for interactive # loop for interactive
while True: while True: