make linux launch.py use XFORMERS_PACKAGE var too; thanks, acncagua

This commit is contained in:
AUTOMATIC 2023-01-30 15:48:09 +03:00
parent ee9fdf7f62
commit 19de2a626b

View File

@ -290,7 +290,7 @@ def prepare_environment():
if not is_installed("xformers"): if not is_installed("xformers"):
exit(0) exit(0)
elif platform.system() == "Linux": elif platform.system() == "Linux":
run_pip("install xformers==0.0.16rc425", "xformers") run_pip(f"install xformers=={xformers_package}", "xformers")
if not is_installed("pyngrok") and ngrok: if not is_installed("pyngrok") and ngrok:
run_pip("install pyngrok", "ngrok") run_pip("install pyngrok", "ngrok")