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")