diff --git a/requirements.txt b/requirements.txt
index 48d9146..9d09350 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -7,4 +7,5 @@ albumentations==1.3.0
 kornia==0.6.9
 wandb==0.13.7
 einops
-torchmetrics
\ No newline at end of file
+torchmetrics
+webdataset
\ No newline at end of file