mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-04-14 16:22:28 +00:00
Update accel_sdxl_gen_img.py
This commit is contained in:
DKnight54
GitHub
@@ -2380,6 +2380,7 @@ def main(args):
|
||||
return images
|
||||
|
||||
# save image
|
||||
'''
|
||||
distributed_state.wait_for_everyone()
|
||||
all_images=gather_object(images)
|
||||
all_global_counter = gather_object(global_counter)
|
||||
@@ -2393,62 +2394,62 @@ def main(args):
|
||||
else:
|
||||
all_init_images = None
|
||||
if distributed_state.is_main_process:
|
||||
|
||||
highres_prefix = ("0" if highres_1st else "1") if highres_fix else ""
|
||||
ds_str = time.strftime("%Y%m%d", time.localtime())
|
||||
ts_str = time.strftime("%H%M%S", time.localtime())
|
||||
for i, (image, globalcount, prompt, negative_prompts, seed, clip_prompt, raw_prompt) in enumerate(
|
||||
zip(all_images, all_global_counter, all_prompts, all_negative_prompts, all_seeds, all_clip_prompts, all_raw_prompts)
|
||||
):
|
||||
if highres_fix:
|
||||
seed -= 1 # record original seed
|
||||
metadata = PngInfo()
|
||||
metadata.add_text("prompt", prompt)
|
||||
metadata.add_text("seed", str(seed))
|
||||
metadata.add_text("sampler", args.sampler)
|
||||
metadata.add_text("steps", str(steps))
|
||||
metadata.add_text("scale", str(scale))
|
||||
if negative_prompt is not None:
|
||||
metadata.add_text("negative-prompt", negative_prompt)
|
||||
if negative_scale is not None:
|
||||
metadata.add_text("negative-scale", str(negative_scale))
|
||||
if clip_prompt is not None:
|
||||
metadata.add_text("clip-prompt", clip_prompt)
|
||||
if raw_prompt is not None:
|
||||
metadata.add_text("raw-prompt", raw_prompt)
|
||||
metadata.add_text("original-height", str(original_height))
|
||||
metadata.add_text("original-width", str(original_width))
|
||||
metadata.add_text("original-height-negative", str(original_height_negative))
|
||||
metadata.add_text("original-width-negative", str(original_width_negative))
|
||||
metadata.add_text("crop-top", str(crop_top))
|
||||
metadata.add_text("crop-left", str(crop_left))
|
||||
|
||||
if args.use_original_file_name and init_images is not None:
|
||||
if type(init_images) is list:
|
||||
fln = os.path.splitext(os.path.basename(init_images[i % len(init_images)].filename))[0] + ".png"
|
||||
else:
|
||||
fln = os.path.splitext(os.path.basename(init_images.filename))[0] + ".png"
|
||||
elif args.sequential_file_name:
|
||||
fln = f"im_{globalcount:02d}_{highres_prefix}{step_first + i + 1:06d}.png"
|
||||
else:
|
||||
fln = f"im_{ds_str}_{ts_str}_{globalcount:02d}_{highres_prefix}{i:03d}_{seed}.png"
|
||||
logger.info(f"Saving image {globalcount}: {fln}\nPrompt: {prompt}")
|
||||
image.save(os.path.join(args.outdir, fln), pnginfo=metadata)
|
||||
|
||||
if not args.no_preview and not highres_1st and args.interactive:
|
||||
try:
|
||||
import cv2
|
||||
|
||||
for prompt, image in zip(prompts, images):
|
||||
cv2.imshow(prompt[:128], np.array(image)[:, :, ::-1]) # プロンプトが長いと死ぬ
|
||||
cv2.waitKey()
|
||||
cv2.destroyAllWindows()
|
||||
except ImportError:
|
||||
logger.error(
|
||||
"opencv-python is not installed, cannot preview / opencv-pythonがインストールされていないためプレビューできません"
|
||||
)
|
||||
'''
|
||||
highres_prefix = ("0" if highres_1st else "1") if highres_fix else ""
|
||||
ds_str = time.strftime("%Y%m%d", time.localtime())
|
||||
ts_str = time.strftime("%H%M%S", time.localtime())
|
||||
for i, (image, globalcount, prompt, negative_prompts, seed, clip_prompt, raw_prompt) in enumerate(
|
||||
zip(all_images, all_global_counter, all_prompts, all_negative_prompts, all_seeds, all_clip_prompts, all_raw_prompts)
|
||||
):
|
||||
if highres_fix:
|
||||
seed -= 1 # record original seed
|
||||
metadata = PngInfo()
|
||||
metadata.add_text("prompt", prompt)
|
||||
metadata.add_text("seed", str(seed))
|
||||
metadata.add_text("sampler", args.sampler)
|
||||
metadata.add_text("steps", str(steps))
|
||||
metadata.add_text("scale", str(scale))
|
||||
if negative_prompt is not None:
|
||||
metadata.add_text("negative-prompt", negative_prompt)
|
||||
if negative_scale is not None:
|
||||
metadata.add_text("negative-scale", str(negative_scale))
|
||||
if clip_prompt is not None:
|
||||
metadata.add_text("clip-prompt", clip_prompt)
|
||||
if raw_prompt is not None:
|
||||
metadata.add_text("raw-prompt", raw_prompt)
|
||||
metadata.add_text("original-height", str(original_height))
|
||||
metadata.add_text("original-width", str(original_width))
|
||||
metadata.add_text("original-height-negative", str(original_height_negative))
|
||||
metadata.add_text("original-width-negative", str(original_width_negative))
|
||||
metadata.add_text("crop-top", str(crop_top))
|
||||
metadata.add_text("crop-left", str(crop_left))
|
||||
|
||||
distributed_state.wait_for_everyone()
|
||||
if args.use_original_file_name and init_images is not None:
|
||||
if type(init_images) is list:
|
||||
fln = os.path.splitext(os.path.basename(init_images[i % len(init_images)].filename))[0] + ".png"
|
||||
else:
|
||||
fln = os.path.splitext(os.path.basename(init_images.filename))[0] + ".png"
|
||||
elif args.sequential_file_name:
|
||||
fln = f"im_{globalcount:02d}_{highres_prefix}{step_first + i + 1:06d}.png"
|
||||
else:
|
||||
fln = f"im_{ds_str}_{ts_str}_{globalcount:02d}_{highres_prefix}{i:03d}_{seed}.png"
|
||||
logger.info(f"Saving image {globalcount}: {fln}\nPrompt: {prompt}")
|
||||
image.save(os.path.join(args.outdir, fln), pnginfo=metadata)
|
||||
|
||||
if not args.no_preview and not highres_1st and args.interactive:
|
||||
try:
|
||||
import cv2
|
||||
|
||||
for prompt, image in zip(prompts, images):
|
||||
cv2.imshow(prompt[:128], np.array(image)[:, :, ::-1]) # プロンプトが長いと死ぬ
|
||||
cv2.waitKey()
|
||||
cv2.destroyAllWindows()
|
||||
except ImportError:
|
||||
logger.error(
|
||||
"opencv-python is not installed, cannot preview / opencv-pythonがインストールされていないためプレビューできません"
|
||||
)
|
||||
|
||||
#distributed_state.wait_for_everyone()
|
||||
return images
|
||||
|
||||
# 画像生成のプロンプトが一周するまでのループ
|
||||
|
||||
Reference in New Issue
Block a user