Separate AutoGPTQ dep to `pip install -e .[auto-gptq]` (#1077)
Browse files* Separate AutoGPTQ dep to `pip install -e .[auto-gptq]`
* Fix code review
- requirements.txt +0 -1
- setup.py +3 -0
requirements.txt
CHANGED
|
@@ -1,5 +1,4 @@
|
|
| 1 |
--extra-index-url https://huggingface.github.io/autogptq-index/whl/cu118/
|
| 2 |
-
auto-gptq==0.5.1
|
| 3 |
packaging
|
| 4 |
peft==0.7.0
|
| 5 |
transformers @ git+https://github.com/huggingface/transformers.git@3cefac1d974db5e2825a0cb2b842883a628be7a0
|
|
|
|
| 1 |
--extra-index-url https://huggingface.github.io/autogptq-index/whl/cu118/
|
|
|
|
| 2 |
packaging
|
| 3 |
peft==0.7.0
|
| 4 |
transformers @ git+https://github.com/huggingface/transformers.git@3cefac1d974db5e2825a0cb2b842883a628be7a0
|
setup.py
CHANGED
|
@@ -61,5 +61,8 @@ setup(
|
|
| 61 |
"mamba-ssm": [
|
| 62 |
"mamba-ssm==1.0.1",
|
| 63 |
],
|
|
|
|
|
|
|
|
|
|
| 64 |
},
|
| 65 |
)
|
|
|
|
| 61 |
"mamba-ssm": [
|
| 62 |
"mamba-ssm==1.0.1",
|
| 63 |
],
|
| 64 |
+
"auto-gptq": [
|
| 65 |
+
"auto-gptq==0.5.1",
|
| 66 |
+
],
|
| 67 |
},
|
| 68 |
)
|