casperhansen
commited on
Commit
•
9be92d1
1
Parent(s):
d7057cc
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 |
)
|