Skip to content

Commit 2e65647

Browse files
authored
Merge pull request #2533 from bmaltais/2515-blip2-captioning-doesnt-see-webp-files-with-patch
Add .webp image extension support to BLIP2 captioning.
2 parents 6e3e53f + 8502989 commit 2e65647

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

kohya_gui/blip2_caption_gui.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ def get_images_in_directory(directory_path):
4242
import os
4343

4444
# List of common image file extensions to look for
45-
image_extensions = [".jpg", ".jpeg", ".png", ".bmp", ".gif"]
45+
image_extensions = [".jpg", ".jpeg", ".png", ".bmp", ".gif", ".webp"]
4646

4747
# Generate a list of image file paths in the directory
4848
image_files = [

0 commit comments

Comments
 (0)