TypeError: sdp_kernel() got an unexpected keyword argument 'enable_mem_efficient'

LHRLAB / ChatKBQA

[ACL 2024] Official resources of "ChatKBQA: A Generate-then-Retrieve Framework for Knowledge Base Question Answering with Fine-tuned Large Language Models".

https://arxiv.org/abs/2310.08975

MIT License

225 stars 21 forks source link

TypeError: sdp_kernel() got an unexpected keyword argument 'enable_mem_efficient' #3

Closed ganlinganlin closed 7 months ago

ganlinganlin commented 7 months ago

I encountered this error during the training of Baichuan2-7b, and after searching for relevant solutions, I found that upgrading torch to version 2.0 was suggested. However, I am curious if there are any alternative solutions without upgrading torch.

TypeError: sdp_kernel() got an unexpected keyword argument 'enable_mem_efficient'

LHRLAB commented 7 months ago

We are using PyTorch version 1.13.1. Other related dependencies have been listed in the requirements.txt file.