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

1 files changed, 1 insertions, 1 deletions

diff --git a/launch.py b/launch.py
index 25909469..c44c48fa 100644
--- 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")