Update convert_llama_weights_to_hf.py (#22525)

db803b69 · kevinpro · GitHub · c6126280 · db803b69
Unverified Commit db803b69 authored Apr 03, 2023 by kevinpro Committed by GitHub Apr 03, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/models/llama/convert_llama_weights_to_hf.py src/transformers/models/llama/convert_llama_weights_to_hf.py +1 -1

No files found.
--- a/src/transformers/models/llama/convert_llama_weights_to_hf.py
+++ b/src/transformers/models/llama/convert_llama_weights_to_hf.py
@@ -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")