diff options
| author | alg-wiki <alg.4chan@gmail.com> | 2022-10-10 21:33:08 +0000 |
|---|---|---|
| committer | alg-wiki <alg.4chan@gmail.com> | 2022-10-10 21:35:07 +0000 |
| commit | 907a88b2d0be320575c2129d8d6a1d4f3a68f9eb (patch) | |
| tree | e3a9b3bf72b5db993466413dc883c561f4a9ff08 /modules/textual_inversion/dataset.py | |
| parent | f0ab972f85b4a185e7ff74b6f325835f1135deff (diff) | |
| download | stable-diffusion-webui-gfx803-907a88b2d0be320575c2129d8d6a1d4f3a68f9eb.tar.gz stable-diffusion-webui-gfx803-907a88b2d0be320575c2129d8d6a1d4f3a68f9eb.tar.bz2 stable-diffusion-webui-gfx803-907a88b2d0be320575c2129d8d6a1d4f3a68f9eb.zip | |
Added .webp .bmp
Diffstat (limited to 'modules/textual_inversion/dataset.py')
| -rw-r--r-- | modules/textual_inversion/dataset.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/textual_inversion/dataset.py b/modules/textual_inversion/dataset.py index d4baf066..0dc54fb7 100644 --- a/modules/textual_inversion/dataset.py +++ b/modules/textual_inversion/dataset.py @@ -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 = []
|