执行转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)