fix up reqs

neuralmagic · Jun 16, 2024 · 69f4dc1 · 69f4dc1
1 parent 6b6bd26
commit 69f4dc1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/requirements-lint.txt b/requirements-lint.txt
@@ -10,5 +10,5 @@ clang-format==18.1.5
 # type checking
 mypy==1.9.0
 types-PyYAML
-types-requests
+types-requests==2.31.0.2
 types-setuptools
diff --git a/requirements-test.txt b/requirements-test.txt
@@ -11,9 +11,11 @@ awscli
 einops # required for MPT
 httpx
 peft
-requests
+requests==2.31  # required for python 3.8 testing
 ray
 sentence-transformers # required for embedding
+optimum     # required for hf gptq baselines
+auto-gptq   # required for hf gptq baselines
 
 # Benchmarking
 aiohttp