Fix docker command to work better with runpod

2026-04-29 18:51:37 +00:00 · 2025-03-27 17:44:46 -06:00
parent 5365200da1
commit 1d5f387f54
4 changed files with 78 additions and 3 deletions
--- a/toolkit/dataloader_mixins.py
+++ b/toolkit/dataloader_mixins.py
@@ -771,7 +771,7 @@ class InpaintControlFileItemDTOMixin:
            img = Image.open(self.inpaint_path)
            # make sure has aplha
            if img.mode != 'RGBA':
-                raise ValueError(f"Image must have alpha channel for inpaint: {self.inpaint_path}")
+                return
            img = exif_transpose(img)
        
            w, h = img.size