UnboundLocalError: local variable 'batch_size' referenced before assignment

[Money8888]

Hi
I have a mistake when using the "model.chat()" function. here are some pictures about it

there may be a mistake happend in " ~/.cache/huggingface/modules/transformers_modules/visualglm-6b/modeling_chatglm.py" 907 line. how can i correct it? thanks

the model and tokenizer have already finished