diff --git a/diffusion/datasets/image_caption.py b/diffusion/datasets/image_caption.py
index 4a41425b..27344a6e 100644
--- a/diffusion/datasets/image_caption.py
+++ b/diffusion/datasets/image_caption.py
@@ -232,9 +232,6 @@ def build_streaming_image_caption_dataloader(
     transform = transforms.Compose(transform)
     assert isinstance(transform, Callable)
 
-    import streaming
-    streaming.base.util.clean_stale_shared_memory()
-
     dataset = StreamingImageCaptionDataset(
         streams=streams,
         tokenizer_name_or_path=tokenizer_name_or_path,