MrYxJ / calculate-flops.pytorch

The calflops is designed to calculate FLOPs、MACs and Parameters in all various neural networks, such as Linear、 CNN、 RNN、 GCN、Transformer(Bert、LlaMA etc Large Language Model)
https://pypi.org/project/calflops/
MIT License
393 stars 14 forks source link

why I can not find transformers.modeling_outputs,Can you tell me your vision of transformers #9

Closed kingofengineer closed 8 months ago

kingofengineer commented 8 months ago

---> 32 from transformers.modeling_outputs import ( 33 CausalLMOutputWithCrossAttentions, 34 Seq2SeqLMOutput, 35 SequenceClassifierOutput, 36 ) ModuleNotFoundError: No module named 'transformers.modeling_outputs'

MrYxJ commented 8 months ago

I recommend vision of transformers >= 4.34.0