fix(#824): can not start the webui on Mac Arm64

This commit is contained in:
bugyalu
2023-05-10 00:06:30 +08:00
parent 4d9fab4984
commit 302e2f9f44
+1 -1
View File
@@ -211,7 +211,7 @@ def check_torch():
xformers_package = os.environ.get('XFORMERS_PACKAGE', 'none')
else:
log.info('Using CPU-only Torch')
torch_command = os.environ.get('TORCH_COMMAND', 'torch torchaudio torchvision')
torch_command = os.environ.get('TORCH_COMMAND', 'torch==2.0.0 torchaudio torchvision')
xformers_package = os.environ.get('XFORMERS_PACKAGE', 'none')
if 'torch' in torch_command and not args.version:
install(torch_command, 'torch torchvision torchaudio')