diff --git a/pyproject.toml b/pyproject.toml
index 9292ae3ab..0946ba6ec 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -29,6 +29,10 @@ dependencies = [
     "openai",
     "tqdm",
     "requests",
+    # Core ML/AI packages
+    "torch",
+    "torchvision",
+    "transformers>=4.57.0,<5.0.0",
     # Data
     "datasets>=2.20.0",
     "pillow",