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

3 years ago · 19de2a626b
parent ee9fdf7f62
commit 19de2a626b
1 changed files with 1 additions and 1 deletions
--- a/launch.py
+++ b/launch.py
@ -290,7 +290,7 @@ def prepare_environment():
                if not is_installed("xformers"):
                    exit(0)
        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:
        run_pip("install pyngrok", "ngrok")