Skip to content

Commit 026cab4

Browse files
authored
Update README.md
1 parent 398870f commit 026cab4

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

README.md

+2-2
Original file line numberDiff line numberDiff line change
@@ -208,8 +208,8 @@ With `export=True`, IPEX model will replace torch linear to ipex linear which pr
208208
Here is the example of how to use IPEX optimized model to generate texts.
209209
### generate
210210
```diff
211-
import torch
212-
from transformers import AutoTokenizer
211+
import torch
212+
from transformers import AutoTokenizer, pipeline
213213
- from transformers import AutoModelForCausalLM
214214
+ from optimum.intel.ipex import IPEXModelForCausalLM
215215

0 commit comments

Comments
 (0)