Pin sentencepiece, add pytorch cuda constraint

aws · Nov 13, 2024 · 506d90e · 506d90e
1 parent f8efc87
commit 506d90e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/build_artifacts/v2/v2.1/v2.1.1/cpu.env.in b/build_artifacts/v2/v2.1/v2.1.1/cpu.env.in
@@ -54,4 +54,4 @@ conda-forge::pyhive[version='>=0.7.0,<0.8.0']
 conda-forge::python-gssapi[version='>=1.9.0,<1.10.0']
 conda-forge::tf-keras[version='>=2.17.0,<2.18.0']
 conda-forge::git-remote-codecommit[version='>=1.16,<1.17.0']
-conda-forge::sentencepiece[version='>=0.1.99,<0.2.0']
+conda-forge::sentencepiece[version='<=0.1.99']
diff --git a/build_artifacts/v2/v2.1/v2.1.1/gpu.env.in b/build_artifacts/v2/v2.1/v2.1.1/gpu.env.in
@@ -12,7 +12,7 @@ conda-forge::amazon-sagemaker-sql-magic[version='>=0.1.3,<0.2.0']
 conda-forge::langchain[version='>=0.2.16,<0.3.0']
 conda-forge::fastapi[version='>=0.115.2,<0.116.0']
 conda-forge::uvicorn[version='>=0.32.0,<0.33.0']
-conda-forge::pytorch[version='>=2.3.1,<2.4.0']
+conda-forge::pytorch[version='>=2.3.1,<2.4.0=*cuda12*']
 conda-forge::tensorflow[version='>=2.17.0,<2.18.0']
 conda-forge::python[version='>=3.11.10,<3.12.0']
 conda-forge::pip[version='>=24.2,<24.3.0']
@@ -54,4 +54,4 @@ conda-forge::pyhive[version='>=0.7.0,<0.8.0']
 conda-forge::python-gssapi[version='>=1.9.0,<1.10.0']
 conda-forge::tf-keras[version='>=2.17.0,<2.18.0']
 conda-forge::git-remote-codecommit[version='>=1.16,<1.17.0']
-conda-forge::sentencepiece[version='>=0.1.99,<0.2.0']
+conda-forge::sentencepiece[version='<=0.1.99']