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

Files changed (2) hide show
  1. requirements.txt +0 -1
  2. 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
  )