Skip to content

Commit fe01c57

Browse files
Update README.md
Co-authored-by: Ella Charlaix <80481427+echarlaix@users.noreply.github.com>
1 parent 4871bba commit fe01c57

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

README.md

+2-1
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,8 @@ from transformers import AutoTokenizer, AutoConfig
5454
+ from optimum.intel.ipex import IPEXModelForCausalLM
5555

5656
config = AutoConfig.from_pretrained("gpt2")
57-
model = IPEXModelForCausalLM.from_pretrained(
57+
- model = AutoModelForCausalLM.from_pretrained(
58+
+ model = IPEXModelForCausalLM.from_pretrained(
5859
"gpt2",
5960
config=config,
6061
torch_dtype=torch.bfloat16,

0 commit comments

Comments
 (0)