pin transformers

rapidsai · Sep 28, 2021 · 6a9987a · 6a9987a
1 parent 0b89459
commit 6a9987a
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/conda/environments/cudf_dev_cuda11.0.yml b/conda/environments/cudf_dev_cuda11.0.yml
@@ -55,7 +55,7 @@ dependencies:
   - protobuf
   - nvtx>=0.2.1
   - cachetools
-  - transformers
+  - transformers<=4.10.3
   - pydata-sphinx-theme
   - pip:
       - git+https://github.com/dask/[email protected]

diff --git a/conda/environments/cudf_dev_cuda11.2.yml b/conda/environments/cudf_dev_cuda11.2.yml
@@ -55,7 +55,7 @@ dependencies:
   - protobuf
   - nvtx>=0.2.1
   - cachetools
-  - transformers
+  - transformers<=4.10.3
   - pydata-sphinx-theme
   - pip:
       - git+https://github.com/dask/[email protected]

diff --git a/python/cudf/requirements/cuda-11.0/dev_requirements.txt b/python/cudf/requirements/cuda-11.0/dev_requirements.txt
@@ -35,6 +35,6 @@ sphinx
 sphinx-copybutton
 sphinx-markdown-tables
 sphinxcontrib-websupport
-transformers
+transformers<=4.10.3
 typing_extensions
 wheel
diff --git a/python/cudf/requirements/cuda-11.2/dev_requirements.txt b/python/cudf/requirements/cuda-11.2/dev_requirements.txt
@@ -35,6 +35,6 @@ sphinx
 sphinx-copybutton
 sphinx-markdown-tables
 sphinxcontrib-websupport
-transformers
+transformers<=4.10.3
 typing_extensions
 wheel
diff --git a/python/cudf/setup.py b/python/cudf/setup.py
@@ -51,7 +51,7 @@
         "hypothesis" "mimesis",
         "pyorc",
         "msgpack",
-        "transformers",
+        "transformers<=4.10.3",
     ]
 }