diff --git a/Dockerfile b/Dockerfile
index f1f2387..233a668 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM kernai/refinery-parent-images:v2.0.0-torch-cpu
+FROM kernai/refinery-parent-images:v2.1.0-torch-cpu
 
 RUN apt-get update && \
     apt-get install --no-install-recommends -y curl libgomp1 && \
diff --git a/requirements.txt b/requirements.txt
index 304e159..551dfbf 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,110 +8,110 @@
 
 annotated-types==0.7.0
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   pydantic
 anyio==4.9.0
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   starlette
 argon2-cffi==25.1.0
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   minio
 argon2-cffi-bindings==21.2.0
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   argon2-cffi
 boto3==1.39.6
-    # via -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    # via -r requirements/torch-cpu-requirements.txt
 botocore==1.39.10
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   boto3
     #   s3transfer
 certifi==2025.7.14
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   minio
     #   requests
 cffi==1.17.1
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   argon2-cffi-bindings
 charset-normalizer==3.4.2
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   requests
 click==8.2.1
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   uvicorn
 fastapi==0.116.1
-    # via -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    # via -r requirements/torch-cpu-requirements.txt
 filelock==3.18.0
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   huggingface-hub
     #   torch
     #   transformers
 fsspec==2025.7.0
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   huggingface-hub
     #   torch
-greenlet==3.2.3
+greenlet==3.2.4
     # via sqlalchemy
 h11==0.16.0
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   uvicorn
 hf-xet==1.1.5
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   huggingface-hub
 huggingface-hub==0.33.4
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   tokenizers
     #   transformers
 idna==3.10
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   anyio
     #   requests
 jinja2==3.1.6
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   torch
 jmespath==1.0.1
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   boto3
     #   botocore
 joblib==1.5.1
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   scikit-learn
     #   scikit-optimize
 lightgbm==4.6.0
     # via -r requirements/requirements.in
 markupsafe==3.0.2
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   jinja2
 minio==7.2.15
-    # via -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    # via -r requirements/torch-cpu-requirements.txt
 mpmath==1.3.0
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   sympy
 networkx==3.5
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   torch
 numpy==1.23.4
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   lightgbm
     #   pandas
     #   scikit-learn
@@ -122,75 +122,75 @@ numpy==1.23.4
     #   xgboost
 packaging==25.0
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   huggingface-hub
     #   transformers
 pandas==1.5.1
-    # via -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    # via -r requirements/torch-cpu-requirements.txt
 psycopg2-binary==2.9.9
-    # via -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    # via -r requirements/torch-cpu-requirements.txt
 pyaml==25.7.0
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   scikit-optimize
 pycparser==2.22
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   cffi
 pycryptodome==3.23.0
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   minio
 pydantic==2.7.4
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   fastapi
 pydantic-core==2.18.4
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   pydantic
 python-dateutil==2.9.0.post0
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   botocore
     #   pandas
 pytz==2025.2
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   pandas
 pyyaml==6.0.2
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   huggingface-hub
     #   pyaml
     #   transformers
 regex==2024.11.6
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   transformers
 requests==2.32.4
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   huggingface-hub
     #   transformers
 s3transfer==0.13.1
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   boto3
 safetensors==0.5.3
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   transformers
 scikit-learn==1.5.2
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   scikit-optimize
     #   sequencelearn
 scikit-optimize==0.9.0
-    # via -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    # via -r requirements/torch-cpu-requirements.txt
 scipy==1.13.1
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   lightgbm
     #   scikit-learn
     #   scikit-optimize
@@ -200,44 +200,44 @@ sequencelearn==0.0.9
     # via -r requirements/requirements.in
 six==1.17.0
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   python-dateutil
 sniffio==1.3.1
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   anyio
 sqlalchemy==1.4.42
-    # via -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    # via -r requirements/torch-cpu-requirements.txt
 starlette==0.47.2
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   fastapi
 sympy==1.14.0
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   torch
 threadpoolctl==3.6.0
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   scikit-learn
 tokenizers==0.21.2
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   transformers
-torch==2.7.1+cpu
+torch==2.8.0+cpu
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   sequencelearn
 tqdm==4.67.1
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   huggingface-hub
     #   transformers
 transformers==4.53.2
-    # via -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    # via -r requirements/torch-cpu-requirements.txt
 typing-extensions==4.14.1
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   anyio
     #   fastapi
     #   huggingface-hub
@@ -248,11 +248,11 @@ typing-extensions==4.14.1
     #   torch
 urllib3==2.5.0
     # via
-    #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    #   -r requirements/torch-cpu-requirements.txt
     #   botocore
     #   minio
     #   requests
 uvicorn==0.35.0
-    # via -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
+    # via -r requirements/torch-cpu-requirements.txt
 xgboost==2.0.1
     # via -r requirements/requirements.in
diff --git a/requirements/torch-cpu-requirements.txt b/requirements/torch-cpu-requirements.txt
index 6c7b67e..761d732 100644
--- a/requirements/torch-cpu-requirements.txt
+++ b/requirements/torch-cpu-requirements.txt
@@ -145,7 +145,7 @@ threadpoolctl==3.6.0
     # via scikit-learn
 tokenizers==0.21.2
     # via transformers
-torch==2.7.1
+torch==2.8.0
     # via -r torch-cpu-requirements.in
 tqdm==4.67.1
     # via