From ec6144fa819653b9e0added32cc8d103c06ef0c0 Mon Sep 17 00:00:00 2001 From: wailovet Date: Tue, 2 Jul 2024 00:45:41 +0800 Subject: [PATCH] 'update' --- mz_train_tools_core_HYDiT.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mz_train_tools_core_HYDiT.py b/mz_train_tools_core_HYDiT.py index e0e0d50..b244919 100644 --- a/mz_train_tools_core_HYDiT.py +++ b/mz_train_tools_core_HYDiT.py @@ -394,7 +394,7 @@ def MZ_HYDiTTrain_call(args={}): writer.writerow(["image_path", "text_zh"]) for filename in full_filenames: - if filename.lower().endswith(".png"): + if filename.lower().endswith(".png") or filename.lower().endswith(".jpg"): image_path = os.path.join(workspace_images_dir, filename) pil_image = Image.open(image_path)