Build doc

huggingface · Dec 19, 2024 · 3041f16 · 3041f16
1 parent b8701ff
commit 3041f16
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 10 deletions.
diff --git a/docs/Dockerfile b/docs/Dockerfile
@@ -8,5 +8,4 @@ RUN python3 -m pip install --no-cache-dir --upgrade pip
 RUN python3 -m pip install --no-cache-dir git+https://github.com/huggingface/doc-builder.git
 
 RUN git clone $clone_url && cd optimum && git checkout $commit_sha
-RUN python3 -m pip install --no-cache-dir ./optimum[onnxruntime,benchmark,quality,exporters-tf,doc-build,diffusers] 
-RUN python3 -m pip install ./optimum[exporters-executorch] --index-url https://download.pytorch.org/whl/cpu
+RUN python3 -m pip install --no-cache-dir ./optimum[onnxruntime,benchmark,quality,exporters-tf,exporters-executorch,doc-build,diffusers] 
diff --git a/setup.py b/setup.py
@@ -76,14 +76,14 @@
     ],
     "exporters-tf": [
         "tensorflow>=2.4,<=2.12.1",
-        "tf2onnx",
-        "onnx",
-        "onnxruntime",
-        "timm",
-        "h5py",
-        "numpy<1.24.0",
-        "datasets<=2.16",
-        "transformers>=4.36",
+        # "tf2onnx",
+        # "onnx",
+        # "onnxruntime",
+        # "timm",
+        # "h5py",
+        # "numpy<1.24.0",
+        # "datasets<=2.16",
+        # "transformers>=4.36",
     ],
     "exporters-executorch": [
         "executorch>=0.4.0",