search

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://doi.org/10.18653/v1/2024.findings-acl.122
MIT License
272 stars 23 forks source link

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

Closed ganlinganlin closed 1 year ago

ganlinganlin commented 1 year 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 1 year ago

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