Spaces:
Running
on
A100
Running
on
A100
Update requirements.txt
Browse filesI 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
```
- 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
|