From 0d7efe3e4b7e057d105d90862cffc4f8cc125d1a Mon Sep 17 00:00:00 2001 From: Xiang Chendong <66510463+Xiang-cd@users.noreply.github.com> Date: Fri, 11 Jul 2025 13:59:41 -0400 Subject: [PATCH] fix gpt2 usage doc (#39351) fix typo of gpt2 doc usage --- docs/source/en/model_doc/gpt2.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/en/model_doc/gpt2.md b/docs/source/en/model_doc/gpt2.md index d67aabfa83..edc32747bd 100644 --- a/docs/source/en/model_doc/gpt2.md +++ b/docs/source/en/model_doc/gpt2.md @@ -57,7 +57,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer model = AutoModelForCausalLM.from_pretrained("openai-community/gpt2", torch_dtype=torch.float16, device_map="auto", attn_implementation="sdpa") tokenizer = AutoTokenizer.from_pretrained("openai-community/gpt2") -input_ids = tokenzier("Hello, I'm a language model". return_tensors="pt").to("cuda") +input_ids = tokenizer("Hello, I'm a language model", return_tensors="pt").to("cuda") output = model.generate(**input_ids, cache_implementation="static") print(tokenizer.decode(output[0], skip_special_tokens=True))