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 index 86ea1ebc..509d8b88 100644 --- 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 index d5591eea..fcebfa5f 100644 --- 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']