default to qlora support, make gptq specific image

2023-05-29 20:34:41 -04:00
parent e43bcc6c4f
commit 6ef96f569b
5 changed files with 25 additions and 49 deletions
--- a/setup.py
+++ b/setup.py
@@ -17,10 +17,10 @@ setup(
    packages=find_packages(),
    install_requires=install_requires,
    extras_require={
-        "int4": [
+        "gptq": [
            "alpaca_lora_4bit @ git+https://github.com/winglian/alpaca_lora_4bit.git@setup_pip",
        ],
-        "int4_triton": [
+        "gptq_triton": [
            "alpaca_lora_4bit[triton] @ git+https://github.com/winglian/alpaca_lora_4bit.git@setup_pip",
        ],
        "extras": [