hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-04 11:12:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge branch 'master' into textual__inversion

Browse Source
This commit is contained in:
alg-wiki
2022-10-11 03:35:28 +08:00
committed by GitHub
parent bc3e183b73 5da1ba0e91
commit f0ab972f85
19 changed files with 1194 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
modules/textual_inversion/textual_inversion.py
View File

@@ -202,6 +202,7 @@ def train_embedding(embedding_name, learn_rate, data_root, log_directory, traini
return embedding, filename
tr_img_len = len([os.path.join(data_root, file_path) for file_path in os.listdir(data_root) if os.path.splitext(file_path.casefold())[1] in extns])
epoch_len = (tr_img_len * num_repeats) + tr_img_len
pbar = tqdm.tqdm(enumerate(ds), total=steps-ititial_step)