diff --git a/requirements-gpu.txt b/requirements-gpu.txt
index 75dd998..41041a3 100644
--- a/requirements-gpu.txt
+++ b/requirements-gpu.txt
@@ -1,5 +1,6 @@
 cython
 ifnude
+albumentations==1.4.3
 insightface==0.7.3
 onnx>=1.14.0
 protobuf>=3.20.2
diff --git a/requirements.txt b/requirements.txt
index 8684bd2..e0be8cf 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,7 @@
 protobuf>=3.20.2
 cython
 ifnude
+albumentations==1.4.3
 insightface==0.7.3
 onnx>=1.14.0
 onnxruntime>=1.15.0