riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 02:31:11 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Update convert_llama_weights_to_hf.py (#22525)

Browse Source
This commit is contained in:
kevinpro 2023-04-03 17:41:39 +08:00 committed by GitHub
parent c612628045
commit db803b6919
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/llama/convert_llama_weights_to_hf.py
View File

@ -34,7 +34,7 @@ python src/transformers/models/llama/convert_llama_weights_to_hf.py \
Thereafter, models can be loaded via:
```py
from transformers import LlamaForCausalLM, LlamaForTokenizer
from transformers import LlamaForCausalLM, LlamaTokenizer
model = LlamaForCausalLM.from_pretrained("/output/path")
tokenizer = LlamaTokenizer.from_pretrained("/output/path")