VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-15 00:32:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update accel_sdxl_gen_img.py

Browse Source
This commit is contained in:
DKnight54
2025-02-02 00:50:03 +08:00
committed by GitHub
parent 75859a6cab
commit 8967e2f014
1 changed files with 4 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
accel_sdxl_gen_img.py
View File

@@ -2469,6 +2469,8 @@ def main(args):
raw_prompt = raw_prompts[pi] if len(raw_prompts) > 1 else raw_prompts[0]
prompt_args = raw_prompt.strip().split(" --")
prompt = prompt_args[0]
if pi == 0 or len(raw_prompts) > 1:
logger.info(f"prompt {prompt_index+1}/{len(prompt_list)}: {prompt}")
for parg in prompt_args[1:]:
try:
@@ -2481,8 +2483,7 @@ def main(args):
except ValueError as ex:
logger.error(f"Exception in parsing / 解析エラー: {parg}")
logger.error(f"{ex}")
if pi == 0 or len(raw_prompts) > 1:
logger.info(f"prompt {prompt_index+1}/{len(prompt_list)}: {prompt}")
if pi == 0:
# parse prompt: if prompt is not changed, skip parsing
@@ -2904,23 +2905,12 @@ def main(args):
if distributed_state.num_processes > 1:
for x in range(len(ext_separated_list_of_batches)):
temp_list = []
logger.info(f"start: ext_separated_list_of_batches[x]: {len(ext_separated_list_of_batches[x])}")
for i in range(distributed_state.num_processes):
temp_list.append(ext_separated_list_of_batches[x][i :: distributed_state.num_processes])
ext_separated_list_of_batches[x] = []
for batches in temp_list:
ext_separated_list_of_batches[x].extend(batches)
logger.info(f"templist: {len(temp_list)}, ext_separated_list_of_batches[x]: {len(ext_separated_list_of_batches[x])}")
logger.info(f"ext_separated_list_of_batches: {len(ext_separated_list_of_batches)}")
count_prompts = 0
for sub_batch_list in ext_separated_list_of_batches:
logger.info(f" sub_batch_list: {len(sub_batch_list)}")
for batches in sub_batch_list:
logger.info(f" batches: {len(batches)}")
count_prompts += len(batches)
logger.info(f"count_prompts: {count_prompts}")
distributed_state.wait_for_everyone()
ext_separated_list_of_batches = gather_object(ext_separated_list_of_batches)
del extinfo