From a8b3f7c9bd2f337dc5e3af9a4287f69fb630be46 Mon Sep 17 00:00:00 2001 From: louistiti Date: Thu, 20 Jun 2024 09:54:16 +0800 Subject: [PATCH] feat(python tcp server): upgrade PyTorch to `2.3.1` (better MPS compatibility) --- scripts/setup/setup-python-dev-env.js | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/scripts/setup/setup-python-dev-env.js b/scripts/setup/setup-python-dev-env.js index 692978ad..da1ce2e6 100644 --- a/scripts/setup/setup-python-dev-env.js +++ b/scripts/setup/setup-python-dev-env.js @@ -152,8 +152,8 @@ SPACY_MODELS.set('fr', { // There is no CUDA support on macOS const commandToExecute = osType === OSTypes.MacOS - ? 'pipenv run pip install --ignore-installed --force-reinstall torch==2.3.0' - : 'pipenv run pip install --ignore-installed --force-reinstall torch==2.3.0 --index-url https://download.pytorch.org/whl/cu121' + ? 'pipenv run pip install --ignore-installed --force-reinstall torch==2.3.1' + : 'pipenv run pip install --ignore-installed --force-reinstall torch==2.3.1 --index-url https://download.pytorch.org/whl/cu121' await command(commandToExecute, { shell: true,