update install-sf.py to handle new 1.05 stable-fast

now supports torch 2.2.2 and 2.3.0
This commit is contained in:
Aptronymist
2024-05-10 21:38:57 -04:00
parent bd3b3bd01c
commit 60ca26198e
+1 -1
View File
@@ -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'