From 576ea88618d4f752e83483b2c30f541c6ba8747d Mon Sep 17 00:00:00 2001 From: Seunghoon Lee Date: Fri, 28 Apr 2023 10:56:00 +0900 Subject: [PATCH] torch 2.0.0 support. --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 99d079463..dda7a4423 100644 --- a/setup.py +++ b/setup.py @@ -200,7 +200,7 @@ def check_torch(): machine = platform.machine() if 'arm' not in machine and 'aarch' not in machine and not args.nodirectml: # torch-directml is available on AMD64 log.info('Using DirectML Backend') - torch_command = os.environ.get('TORCH_COMMAND', 'torch==1.13.1 torchvision==0.14.1 torch-directml') + torch_command = os.environ.get('TORCH_COMMAND', 'torch==2.0.0 torchvision torch-directml') xformers_package = os.environ.get('XFORMERS_PACKAGE', 'none') else: log.info('Using CPU-only Torch')