diff --git a/cli/install-sf.py b/cli/install-sf.py index 2febb46be..e60b551b9 100755 --- a/cli/install-sf.py +++ b/cli/install-sf.py @@ -3,7 +3,7 @@ import os import re import sys -torch_supported = ['211', '212','220','221'] +torch_supported = ['211', '212','220','221','222','230'] cuda_supported = ['cu118', 'cu121'] python_supported = ['39', '310', '311'] repo_url = 'https://github.com/chengzeyi/stable-fast'