From 60ca26198e9ad1511e0795f4a33a9fc4ddc80a58 Mon Sep 17 00:00:00 2001 From: Aptronymist <108482020+Aptronymist@users.noreply.github.com> Date: Fri, 10 May 2024 21:38:57 -0400 Subject: [PATCH] update install-sf.py to handle new 1.05 stable-fast now supports torch 2.2.2 and 2.3.0 --- cli/install-sf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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'