Fix modality for image inputs (#1592)

This commit is contained in:
Lianmin Zheng
2024-10-06 15:43:32 -07:00
committed by GitHub
parent f7cce751f9
commit 91877a9f9c

View File

@@ -144,7 +144,7 @@ class ImageInputs:
]
ret.image_sizes = obj["image_sizes"]
# Only when pixel values is not None we have modalities
ret.modalities = obj["modalities"]
ret.modalities = obj["modalities"] or ["image"]
return ret