docker layer caching, build w axolotl from base build

2023-05-07 17:16:05 -04:00
parent 0c46806ae2
commit 990bec63e6
6 changed files with 27 additions and 19 deletions
--- a/setup.py
+++ b/setup.py
@@ -23,5 +23,9 @@ setup(
        'int4_triton': [
            "alpaca_lora_4bit[triton] @ git+https://github.com/winglian/alpaca_lora_4bit.git@setup_pip",
        ],
+        'extras': [
+            'flash-attn',
+            'deepspeed',
+        ]
    },
 )