推理报错：assert len(logits) == num_tokens and logits[0].shape[0] == bsz #17

执行infer_A_md_brief.sh时，在DepictQA/src/model/depictqa.py文件484行处报错，该处代码为：assert len(logits) == num_tokens and logits[0].shape[0] == bsz
打印几个长度分别为：
len(logits):12
num_tokens:11
logits[0].shape[0]:1
bsz:1
这个正常么？还是我哪里操作不当？