Baichuan-7B 无法从'transformers.pytorch_utils'导入名称'id_tensor_storage',

c2e8gylq  于 2个月前  发布在  其他
关注(0)|答案(1)|浏览(29)

必需的先决条件

问题

运行

model = AutoModelForCausalLM.from_pretrained("baichuan-inc/baichuan-7B", device_map="auto", trust_remote_code=True)

时出现报错:

cannot import name 'id_tensor_storage' from 'transformers.pytorch_utils' (/home/pai/lib/python3.9/site-packages/transformers/pytorch_utils.py)

transformers==4.29.1
请问是什么原因呢,应该不是transformers的版本低导致的

检查清单

  • 我已提供所有相关且必要的信息。
  • 我已为这个问题选择了合适的标题。

相关问题