make mlflow optional (#1317)

* make mlflow optional * fix xformers don't patch swiglu if xformers not working fix the check for xformers swiglu * fix install of xformers with extra index url for docker builds * fix docker build arg quoting
2024-02-26 11:41:33 -05:00
parent 5cf226e177
commit 5894f0e57e
12 changed files with 86 additions and 41 deletions
--- a/setup.py
+++ b/setup.py
@@ -82,5 +82,8 @@ setup(
        "auto-gptq": [
            "auto-gptq==0.5.1",
        ],
+        "mlflow": [
+            "mlflow",
+        ],
    },
 )