KnowLM

KnowLM copied to clipboard

在复现信息抽取的结果遇到一系列问题

2

严格按照1.1环境配置来配置环境 首先遇到TypeError，随后更新了transformers包  然后执行命令python examples/generate_lora.py --base_model zjunlp/knowlm-13b-zhixi --run_ie_cases 遇到显存不足的问题  然后修改命令为CUDA_VISIBLE_DEVICES=0,1 python examples/generate_lora.py --base_model zjunlp/knowlm-13b-zhixi --run_ie_cases 也是显示同样的信息，似乎1号卡并没有用上。。服务器有两张titan，显存加起来有48G 再修改命令为CUDA_VISIBLE_DEVICES=0,1 python examples/generate_lora.py --base_model zjunlp/knowlm-13b-zhixi --run_ie_cases --multi_gpu ，出现如下结果  若按照提示信息执行 pip install protobuf==3.20.0，运行命令又会报如下错误...

xj120