Skip to content

Commit 7570d71

Browse files
committed
fix readme
1 parent 8a49f9c commit 7570d71

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

README.md

+1-3
Original file line numberDiff line numberDiff line change
@@ -49,15 +49,13 @@ Here is the example of how to use IPEX optimized model to generate texts.
4949
### generate
5050
```diff
5151
import torch
52-
from transformers import AutoTokenizer, AutoConfig
52+
from transformers import AutoTokenizer
5353
- from transformers import AutoModelForCausalLM
5454
+ from optimum.intel.ipex import IPEXModelForCausalLM
5555

56-
config = AutoConfig.from_pretrained("gpt2")
5756
- model = AutoModelForCausalLM.from_pretrained(
5857
+ model = IPEXModelForCausalLM.from_pretrained(
5958
"gpt2",
60-
config=config,
6159
torch_dtype=torch.bfloat16,
6260
export=True,
6361
)

0 commit comments

Comments
 (0)