I'm running the script/bench_throughput.sh in the e2e branch, while the error occured:
'''
Traceback (most recent call last):
File "BitDecoding/evaluation/bench_throughput.py", line 7, in
from llama import LlamaForCausalLM
File "BitDecoding/evaluation/llama.py", line 43, in
from transformers.utils import (
ImportError: cannot import name 'LossKwargs' from 'transformers.utils' (/usr/local/lib/python3.10/dist-packages/transformers/utils/init.py)
'''
so, which version of transformers do you use ?