From 34a5ef8c15fc925b2263f4fc951b35a376cadffd Mon Sep 17 00:00:00 2001
From: boopage <83878425+boopage@users.noreply.github.com>
Date: Wed, 5 Mar 2025 11:13:11 +0100
Subject: [PATCH] Update train_wan_t2v.py

Included .jpeg extension for image type detection, preventing an error trying to the read image as a video format
---
 examples/wanvideo/train_wan_t2v.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/wanvideo/train_wan_t2v.py b/examples/wanvideo/train_wan_t2v.py
index 45fbe90..c695622 100644
--- a/examples/wanvideo/train_wan_t2v.py
+++ b/examples/wanvideo/train_wan_t2v.py
@@ -70,7 +70,7 @@ class TextVideoDataset(torch.utils.data.Dataset):
     
     def is_image(self, file_path):
         file_ext_name = file_path.split(".")[-1]
-        if file_ext_name.lower() in ["jpg", "png", "webp"]:
+        if file_ext_name.lower() in ["jpg", "jpeg", "png", "webp"]:
             return True
         return False