Merge pull request #8 from kijai/main

Fix mask output
2023-09-24 12:27:07 -05:00
parent 2666ef7aeb ea33b45313
commit 56c3e85407
1 changed files with 1 additions and 1 deletions
@@ -562,7 +562,7 @@ class LoadImagesFromDirectory:
        if len(images) == 0:
            raise FileNotFoundError(f"No images could be loaded from directory '{directory}'.")

-        return (torch.cat(images, dim=0), torch.cat(masks, dim=0), image_count)
+        return (torch.cat(images, dim=0), torch.stack(masks, dim=0), image_count)