diff --git a/requirements/base.txt b/requirements/base.txt
index 19cec0e..58b90c0 100644
--- a/requirements/base.txt
+++ b/requirements/base.txt
@@ -7,6 +7,7 @@ invisible-watermark==0.2.0
 accelerate==0.18.0
 transformers==4.28.1
 sentencepiece~=0.1
+pydantic==2.3.0
 omegaconf
 safetensors