search

THUDM / SwissArmyTransformer

SwissArmyTransformer is a flexible and powerful library to develop your own Transformer variants.
https://THUDM.github.io/SwissArmyTransformer
Apache License 2.0
1.02k stars 97 forks source link

UnboundLocalError: local variable 'batch_size' referenced before assignment #186

Open Money8888 opened 1 week ago

Money8888 commented 1 week ago

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

image image

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

Money8888 commented 1 week ago

the model and tokenizer have already finished

image