Skip to content

Commit 35cf1d2

Browse files
authored
Update readme (#1066)
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
1 parent bb51139 commit 35cf1d2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

README.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -168,7 +168,7 @@ To load your IPEX model, you can just replace your `AutoModelForXxx` class with
168168

169169
model_id = "gpt2"
170170
- model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16)
171-
+ model = IPEXModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16, export=True)
171+
+ model = IPEXModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16)
172172
tokenizer = AutoTokenizer.from_pretrained(model_id)
173173
pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
174174
results = pipe("He's a dreadful magician and")

0 commit comments

Comments
 (0)