I Failed to execute the shell hf_bart_export.py

bytedance / lightseq

LightSeq: A High Performance Library for Sequence Processing and Generation

Other

3.22k stars 329 forks source link

I Failed to execute the shell hf_bart_export.py #400

Closed YINGPENGZH closed 2 years ago

YINGPENGZH commented 2 years ago

Traceback (most recent call last): File "ls_bart.py", line 82, in main() File "ls_bart.py", line 49, in main ls_model = lsi.Transformer("./lightseq_bart_base.hdf5", 128) RuntimeError: violate dim_per_head % 2 = 0

Could you please tell me how to execute the python shell?