Modify torch version requirement to adapt torch 2.0

2023-06-01 14:30:51 +08:00 · 2023-06-01 14:30:51 +08:00 · 46503c35dd
parent 281b33f362
commit 46503c35dd
1 changed files with 1 additions and 1 deletions
--- a/colossalai/cli/launcher/run.py
+++ b/colossalai/cli/launcher/run.py
@ -154,7 +154,7 @@ def get_launch_command(
        extra_launch_args = dict()

    torch_version = version.parse(torch.__version__)
-    assert torch_version.major == 1
+    assert torch_version.major >= 1

    if torch_version.minor < 9:
        cmd = [