diff --git a/environments/huggingface.yml b/environments/huggingface.yml
index ba623854..e027a8ea 100644
--- a/environments/huggingface.yml
+++ b/environments/huggingface.yml
@@ -33,7 +33,7 @@ dependencies:
     - lupa==1.10
     - transformers[sentencepiece]==4.33.1
     - huggingface_hub==0.16.4
-    - optimum[onnx]==1.12.0
+    - optimum[onnxruntime]==1.12.0
     - safetensors==0.3.3
     - accelerate==0.21.0
     - git+https://github.com/VE-FORBRYDERNE/mkultra
diff --git a/environments/ipex.yml b/environments/ipex.yml
index 55ea51cb..4c2dcb52 100644
--- a/environments/ipex.yml
+++ b/environments/ipex.yml
@@ -32,7 +32,7 @@ dependencies:
     - lupa==1.10
     - transformers[sentencepiece]==4.33.1
     - huggingface_hub==0.16.4
-    - optimum[onnx]==1.12.0
+    - optimum[onnxruntime]==1.12.0
     - safetensors==0.3.3
     - accelerate==0.20.3
     - git+https://github.com/VE-FORBRYDERNE/mkultra
diff --git a/environments/rocm.yml b/environments/rocm.yml
index 73c9be2e..34974726 100644
--- a/environments/rocm.yml
+++ b/environments/rocm.yml
@@ -31,7 +31,7 @@ dependencies:
     - lupa==1.10
     - transformers[sentencepiece]==4.33.1
     - huggingface_hub==0.16.4
-    - optimum[onnx]==1.12.0
+    - optimum[onnxruntime]==1.12.0
     - safetensors==0.3.3
     - accelerate==0.21.0
     - git+https://github.com/VE-FORBRYDERNE/mkultra
diff --git a/requirements.txt b/requirements.txt
index ce523629..b6026bfd 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,6 @@
 transformers[sentencepiece]==4.33.1
 huggingface_hub==0.16.4
-optimum[onnx]==1.12.0
+optimum[onnxruntime]==1.12.0
 safetensors==0.3.3
 Flask==2.2.3
 Flask-SocketIO==5.3.2