diff --git a/requirements.txt b/requirements.txt
index 484fb298..c251a40a 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -13,7 +13,8 @@ invisible-watermark
 einops
 accelerate
 toml
-albumentations
+albumentations==1.4.15
+albucore==0.0.16
 pydantic
 omegaconf
 k-diffusion