leon-ai/leon
1
1
Fork 0
mirror of https://github.com/leon-ai/leon.git synced 2024-08-17 06:00:33 +03:00
Code Issues Projects Releases Wiki Activity

feat(python tcp server): upgrade PyTorch to 2.3.1 (better MPS compatibility)

Browse Source
This commit is contained in:
louistiti 2024-06-20 09:54:16 +08:00
parent 963660db19
commit a8b3f7c9bd
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/setup/setup-python-dev-env.js
View File

@ -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,