Yu Xin

能提供一个Patch吗，直接换了ChatGLM，attention mask那个地方报错了。不知道怎么搞. ChatGLM直接出来的attention_mask attention_mask.shape:torch.Size([12, 1, 714, 714]) [batch*2, 1, seq_length,seq_length] ``` Traceback (most recent call last): File "/home/jovyan/ws/nbce/test.py", line 114, in generate(1000) File "/opt/conda/envs/alpaca/lib/python3.10/site-packages/torch/autograd/grad_mode.py", line 27, in decorate_context return func(*args,...