StableDiffusion/KohyaSS
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix UI logic

Browse Source
This commit is contained in:
bmaltais 2022-12-15 18:44:02 -05:00
parent 2f91fd24ae
commit a7cd798abf
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
dreambooth_gui.py
View File

@ -490,7 +490,7 @@ def dreambooth_folder_preparation(
shutil.copytree(util_training_images_dir_input, training_dir) shutil.copytree(util_training_images_dir_input, training_dir)
# Create the regularization_dir path # Create the regularization_dir path
if (not len(util_regularization_images_prompt_input) if (not (util_regularization_images_prompt_input == "")
or not util_regularization_images_repeat_input > 0): or not util_regularization_images_repeat_input > 0):
print( print(
"Regularization images directory or repeats is missing... not copying regularisation images..." "Regularization images directory or repeats is missing... not copying regularisation images..."
@ -520,7 +520,10 @@ def dreambooth_folder_preparation(
def copy_info_to_Directories_tab(training_folder): def copy_info_to_Directories_tab(training_folder):
img_folder = os.path.join(training_folder, "img") img_folder = os.path.join(training_folder, "img")
reg_folder = os.path.join(training_folder, "reg") if os.path.exists(os.path.join(training_folder, "reg")):
reg_folder = os.path.join(training_folder, "reg")
else:
reg_folder = ""
model_folder = os.path.join(training_folder, "model") model_folder = os.path.join(training_folder, "model")
log_folder = os.path.join(training_folder, "log") log_folder = os.path.join(training_folder, "log")