support sys.argv --listen --share --port
This commit is contained in:
lllyasviel
2023-08-14 06:56:23 -07:00
committed by GitHub
parent 50708f3d22
commit 7900480360
5 changed files with 25 additions and 4 deletions
+8 -1
View File
@@ -72,9 +72,16 @@ def download_models():
return
def cuda_malloc():
argv = sys.argv
sys.argv = [sys.argv[0]]
import cuda_malloc
sys.argv = argv
prepare_environment()
import cuda_malloc
cuda_malloc()
download_models()