Pin sentencepiece, add pytorch cuda constraint

aws · Nov 13, 2024 · ea5ed3a · ea5ed3a
1 parent 137658d
commit ea5ed3a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/build_artifacts/v2/v2.0/v2.0.1/cpu.env.in b/build_artifacts/v2/v2.0/v2.0.1/cpu.env.in
@@ -53,3 +53,4 @@ conda-forge::jupyterlab-git[version='>=0.50.1,<0.51.0']
 conda-forge::pyhive[version='>=0.7.0,<0.8.0']
 conda-forge::python-gssapi[version='>=1.8.3,<1.9.0']
 conda-forge::tf-keras[version='>=2.17.0,<2.18.0']
+conda-forge::sentencepiece[version='<=0.1.99']
diff --git a/build_artifacts/v2/v2.0/v2.0.1/gpu.env.in b/build_artifacts/v2/v2.0/v2.0.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.5,<0.3.0']
 conda-forge::fastapi[version='>=0.110.3,<0.111.0']
 conda-forge::uvicorn[version='>=0.30.6,<0.31.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.9,<3.12.0']
 conda-forge::pip[version='>=24.2,<24.3.0']
@@ -53,3 +53,5 @@ conda-forge::jupyterlab-git[version='>=0.50.1,<0.51.0']
 conda-forge::pyhive[version='>=0.7.0,<0.8.0']
 conda-forge::python-gssapi[version='>=1.8.3,<1.9.0']
 conda-forge::tf-keras[version='>=2.17.0,<2.18.0']
+conda-forge::sentencepiece[version='<=0.1.99']
+