Added .webp .bmp

This commit is contained in:
alg-wiki
2022-10-11 06:33:08 +09:00
parent f0ab972f85
commit 907a88b2d0
3 changed files with 3 additions and 3 deletions

View File

@@ -161,7 +161,7 @@ def train_embedding(embedding_name, learn_rate, data_root, log_directory, traini
shared.state.textinfo = "Initializing textual inversion training..."
shared.state.job_count = steps
extns = [".jpg",".jpeg",".png"]
extns = [".jpg",".jpeg",".png",".webp",".bmp"]
filename = os.path.join(shared.cmd_opts.embeddings_dir, f'{embedding_name}.pt')