PKU-YuanGroup / ChatLaw

ChatLaw:A Powerful LLM Tailored for Chinese Legal. 中文法律大模型

Home Page:https://chatlaw.cloud/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

执行转llama-13b 为 llama-13b-hf 时,报LlamaConfig 找不到问题

nuaabuaa07 opened this issue · comments

执行转llama-13b 为 llama-13b-hf 时,报LlamaConfig 找不到

root@kml-dtmachine-10688-prod:/home/jinzhiliang/webdownload/chatgml/model2download# python convert_llama_weights_to_hf.py --input_dir ./llama-13b --model_size 13B --output_dir ./llama-13b-hf
Traceback (most recent call last):
File "convert_llama_weights_to_hf.py", line 23, in
from transformers import LlamaConfig, LlamaForCausalLM, LlamaTokenizer
ImportError: cannot import name 'LlamaConfig' from 'transformers' (/usr/local/lib/python3.8/dist-packages/transformers/init.py)