marksaroufim commited on
Commit
1a175c1
·
verified ·
1 Parent(s): 23210c5

Update requirements.txt

Browse files

I get this error otherwise

```
File "/home/marksaroufim/.conda/envs/nv/lib/python3.10/site-packages/transformers/quantizers/auto.py", line 198, in merge_quantization_configs
quantization_config = AutoQuantizationConfig.from_dict(quantization_config)
File "/home/marksaroufim/.conda/envs/nv/lib/python3.10/site-packages/transformers/quantizers/auto.py", line 128, in from_dict
return target_cls.from_dict(quantization_config_dict)
File "/home/marksaroufim/.conda/envs/nv/lib/python3.10/site-packages/transformers/utils/quantization_config.py", line 1647, in from_dict
len(quant_type) == 1 and "default" in quant_type
AssertionError: Expected only one key 'default' in quant_type dictionary
```

Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
  git+https://github.com/huggingface/transformers.git@main#egg=transformers
2
  accelerate
3
- torchao
4
  huggingface-hub
5
  https://gradio-builds.s3.amazonaws.com/4485dd46a8e4b3f5b35e42d52f291b72fdc1a952/gradio-4.39.0-py3-none-any.whl
 
1
  git+https://github.com/huggingface/transformers.git@main#egg=transformers
2
  accelerate
3
+ git+https://github.com/pytorch/ao.git@main#egg=torchao
4
  huggingface-hub
5
  https://gradio-builds.s3.amazonaws.com/4485dd46a8e4b3f5b35e42d52f291b72fdc1a952/gradio-4.39.0-py3-none-any.whl