mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2025-12-21 23:00:59 -08:00
Added .webp .bmp
This commit is contained in:
@@ -22,7 +22,7 @@ class PersonalizedBase(Dataset):
|
||||
self.width = width
|
||||
self.height = height
|
||||
self.flip = transforms.RandomHorizontalFlip(p=flip_p)
|
||||
self.extns = [".jpg",".jpeg",".png"]
|
||||
self.extns = [".jpg",".jpeg",".png",".webp",".bmp"]
|
||||
|
||||
self.dataset = []
|
||||
|
||||
|
||||
Reference in New Issue
Block a user