Merge pull request #140 from invoke-ai/fix-image-caption-example-bug

Fix pydantic `ValidationError` while creating dataset via the UI.
invoke-ai · Jun 5, 2024 · 734a442 · 734a442
2 parents 36e0ac7 + b7775fd
commit 734a442
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/invoke_training/_shared/data/datasets/image_caption_jsonl_dataset.py b/src/invoke_training/_shared/data/datasets/image_caption_jsonl_dataset.py
@@ -15,7 +15,7 @@
 
 class ImageCaptionExample(BaseModel):
     image_path: str
-    mask_path: str | None
+    mask_path: str | None = None
     caption: str