diff --git a/requirements.txt b/requirements.txt
index c279ff0..c74e46d 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -11,3 +11,4 @@ mlflow==1.23.1
 protobuf==3.19.3
 boto3==1.23.9
 huggingface-hub==0.4.0
+pytorch-lightning==1.5.10