繰り返し回数のないディレクトリの名前表示修正

This commit is contained in:
u-haru
2023-03-31 02:26:54 +09:00
parent 3032a47af4
commit 94441fa746

View File

@@ -445,7 +445,7 @@ def generate_dreambooth_subsets_config_by_subdirs(train_data_dir: Optional[str]
try: try:
n_repeats = int(tokens[0]) n_repeats = int(tokens[0])
except ValueError as e: except ValueError as e:
print(f"ignore directory without repeats / 繰り返し回数のないディレクトリを無視します: {dir}") print(f"ignore directory without repeats / 繰り返し回数のないディレクトリを無視します: {name}")
return 0, "" return 0, ""
caption_by_folder = '_'.join(tokens[1:]) caption_by_folder = '_'.join(tokens[1:])
return n_repeats, caption_by_folder return n_repeats, caption_by_folder