用自己的数据集进行推理程序阻塞了，batch_size已经设置为1 #16

作者您好，我在自己数据集上进行推理，进程阻塞在这行代码不继续执行，不知道是什么原因。

/root/project2024/DepictQA/src/infer.py(56)main()
-> texts, output_ids, probs, confidences = model.generate(
(Pdb) n

推理脚本为：
#!/bin/bash
src_dir=../../src/
export PYTHONPATH=$src_dir:$PYTHONPATH
export CUDA_VISIBLE_DEVICES=$1

python $src_dir/infer.py
--meta_path IQA_data20240924/ref_data/white_balance_label/3_test.json
--dataset_name livemd_AB
--task_name quality_compare_noref
--batch_size 1 \

模型ckpt.pt 是在自己数据集上训练得到的
另外很奇怪的一点，ctrl+c 杀死推理进程时，则会得到一个answer,
譬如：
{"id": "000007", "text": "{"white_balance": "8"}\n ", "output_id": null, "prob": null, "confidence": null}
但是后续仍继续阻塞。