Modify torch version requirement to adapt torch 2.0

46503c35 · Maruyama_Aya · 281b33f3 · 46503c35
Commit 46503c35 authored Jun 01, 2023 by Maruyama_Aya
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

colossalai/cli/launcher/run.py colossalai/cli/launcher/run.py +1 -1

No files found.
--- 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 = [