huggingface / nn_pruning

Prune a model while finetuning or training.
Apache License 2.0
394 stars 58 forks source link

AttributeError: 'GraphModule' object has no attribute 'config' #35

Open LYF915 opened 2 years ago

LYF915 commented 2 years ago

when test quantization, it raises errors. May I ask if anyone has encountered this problem?

pytorch==3.8.1 transformers==4.7.0

截屏2022-02-28 下午5 55 01