mirror of
https://github.com/leon-ai/leon.git
synced 2024-08-17 06:00:33 +03:00
feat(python tcp server): upgrade PyTorch to 2.3.1
(better MPS compatibility)
This commit is contained in:
parent
963660db19
commit
a8b3f7c9bd
@ -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,
|
||||
|
Loading…
Reference in New Issue
Block a user