diff --git a/setup.py b/setup.py
index dee0576..a6d0331 100644
--- a/setup.py
+++ b/setup.py
@@ -11,6 +11,7 @@ setup(
         "wandb",
         "torchvision",
         "inflect==6.0.4",
+        "pydantic==1.10.9",
         "transformers==4.30.2",
     ],
 )