diff --git a/requirements/base.txt b/requirements/base.txt
index 1dc5a65..19dd467 100644
--- a/requirements/base.txt
+++ b/requirements/base.txt
@@ -32,6 +32,8 @@ click==8.1.3
     #   nltk
     #   unstructured-api-tools
     #   uvicorn
+cmake==3.26.3
+    # via triton
 commonmark==0.9.1
     # via rich
 defusedxml==0.7.1
@@ -49,6 +51,7 @@ filelock==3.10.7
     #   huggingface-hub
     #   torch
     #   transformers
+    #   triton
 h11==0.14.0
     # via
     #   httpcore
@@ -66,13 +69,6 @@ idna==3.4
     #   anyio
     #   requests
     #   rfc3986
-importlib-metadata==6.1.0
-    # via
-    #   jupyter-client
-    #   markdown
-    #   nbconvert
-importlib-resources==5.12.0
-    # via jsonschema
 jinja2==3.1.2
     # via
     #   nbconvert
@@ -92,6 +88,8 @@ jupyter-core==5.3.0
     #   nbformat
 jupyterlab-pygments==0.2.2
     # via nbconvert
+lit==16.0.5
+    # via triton
 lxml==4.9.2
     # via
     #   python-docx
@@ -132,6 +130,31 @@ numpy==1.23.5
     #   argilla
     #   pandas
     #   transformers
+nvidia-cublas-cu11==11.10.3.66
+    # via
+    #   nvidia-cudnn-cu11
+    #   nvidia-cusolver-cu11
+    #   torch
+nvidia-cuda-cupti-cu11==11.7.101
+    # via torch
+nvidia-cuda-nvrtc-cu11==11.7.99
+    # via torch
+nvidia-cuda-runtime-cu11==11.7.99
+    # via torch
+nvidia-cudnn-cu11==8.5.0.96
+    # via torch
+nvidia-cufft-cu11==10.9.0.58
+    # via torch
+nvidia-curand-cu11==10.2.10.91
+    # via torch
+nvidia-cusolver-cu11==11.4.0.1
+    # via torch
+nvidia-cusparse-cu11==11.7.4.91
+    # via torch
+nvidia-nccl-cu11==2.14.3
+    # via torch
+nvidia-nvtx-cu11==11.7.91
+    # via torch
 olefile==0.46
     # via msg-parser
 openpyxl==3.1.2
@@ -152,8 +175,6 @@ pillow==9.4.0
     # via
     #   python-pptx
     #   unstructured
-pkgutil-resolve-name==1.3.10
-    # via jsonschema
 platformdirs==3.2.0
     # via jupyter-core
 pydantic==1.10.7
@@ -192,14 +213,14 @@ pyyaml==6.0
 pyzmq==25.0.2
     # via jupyter-client
 ratelimit==2.2.1
-    # via -r requirements/base.in
+    # via -r base.in
 regex==2023.3.23
     # via
     #   nltk
     #   transformers
-requests==2.28.2
+requests==2.31.0
     # via
-    #   -r requirements/base.in
+    #   -r base.in
     #   huggingface-hub
     #   transformers
     #   unstructured
@@ -226,10 +247,10 @@ tinycss2==1.2.1
     # via nbconvert
 tokenizers==0.13.2
     # via transformers
-tomli==2.0.1
-    # via mypy
 torch==2.0.0
-    # via -r requirements/base.in
+    # via
+    #   -r base.in
+    #   triton
 tornado==6.2
     # via jupyter-client
 tqdm==4.65.0
@@ -246,7 +267,9 @@ traitlets==5.9.0
     #   nbconvert
     #   nbformat
 transformers==4.27.4
-    # via -r requirements/base.in
+    # via -r base.in
+triton==2.0.0
+    # via torch
 types-requests==2.28.11.17
     # via unstructured-api-tools
 types-ujson==5.7.0.1
@@ -258,13 +281,11 @@ typing-extensions==4.5.0
     #   huggingface-hub
     #   mypy
     #   pydantic
-    #   rich
-    #   starlette
     #   torch
 unstructured==0.5.8
-    # via -r requirements/base.in
+    # via -r base.in
 unstructured-api-tools==0.8.0
-    # via -r requirements/base.in
+    # via -r base.in
 urllib3==1.26.15
     # via requests
 uvicorn[standard]==0.21.1
@@ -279,13 +300,17 @@ webencodings==0.5.1
     #   tinycss2
 websockets==10.4
     # via uvicorn
+wheel==0.40.0
+    # via
+    #   nvidia-cublas-cu11
+    #   nvidia-cuda-cupti-cu11
+    #   nvidia-cuda-runtime-cu11
+    #   nvidia-curand-cu11
+    #   nvidia-cusparse-cu11
+    #   nvidia-nvtx-cu11
 wrapt==1.14.1
     # via
     #   argilla
     #   deprecated
 xlsxwriter==3.0.9
     # via python-pptx
-zipp==3.15.0
-    # via
-    #   importlib-metadata
-    #   importlib-resources