Bump torch & torch-directml.

lshqqytiger · Oct 22, 2024 · ae5ff7a · ae5ff7a
1 parent efddd05
commit ae5ff7a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/launch_utils.py b/modules/launch_utils.py
@@ -444,7 +444,7 @@ def prepare_environment():
         backend = "directml"
         torch_command = os.environ.get(
             "TORCH_COMMAND",
-            "pip install torch==2.3.1 torchvision torch-directml",
+            "pip install torch==2.4.1 torchvision torch-directml",
         )
         args.skip_python_version_check = True
     elif args.use_zluda: