diff --git a/examples/contrastive-image-text/clip_media_pipe.py b/examples/contrastive-image-text/clip_media_pipe.py index 574837e38f..a4248959c7 100755 --- a/examples/contrastive-image-text/clip_media_pipe.py +++ b/examples/contrastive-image-text/clip_media_pipe.py @@ -36,7 +36,7 @@ pass read_image_text_from_dataset_params = { - "label_dtype": dtype.UINT64, + "label_dtype": dtype.UINT32, "dataset": None, }