openvinotoolkit/stable-diffusion-webui
1
1
Fork 0
mirror of https://github.com/openvinotoolkit/stable-diffusion-webui.git synced 2024-12-15 15:13:45 +03:00
Code Issues Packages Projects Releases Wiki Activity

Added .webp .bmp

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/textual_inversion/dataset.py
View File

@ -22,7 +22,7 @@ class PersonalizedBase(Dataset):
self.width = width self.width = width
self.height = height self.height = height
self.flip = transforms.RandomHorizontalFlip(p=flip_p) self.flip = transforms.RandomHorizontalFlip(p=flip_p)
self.extns = [".jpg",".jpeg",".png"] self.extns = [".jpg",".jpeg",".png",".webp",".bmp"]
self.dataset = [] self.dataset = []

2
modules/textual_inversion/preprocess.py
View File

@ -12,7 +12,7 @@ def preprocess(process_src, process_dst, process_width, process_height, process_
height = process_height height = process_height
src = os.path.abspath(process_src) src = os.path.abspath(process_src)
dst = os.path.abspath(process_dst) dst = os.path.abspath(process_dst)
extns = [".jpg",".jpeg",".png"] extns = [".jpg",".jpeg",".png",".webp",".bmp"]
assert src != dst, 'same directory specified as source and destination' assert src != dst, 'same directory specified as source and destination'

2
modules/textual_inversion/textual_inversion.py
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.textinfo = "Initializing textual inversion training..."
shared.state.job_count = steps 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') filename = os.path.join(shared.cmd_opts.embeddings_dir, f'{embedding_name}.pt')