We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 4871bba commit fe01c57Copy full SHA for fe01c57
README.md
@@ -54,7 +54,8 @@ from transformers import AutoTokenizer, AutoConfig
54
+ from optimum.intel.ipex import IPEXModelForCausalLM
55
56
config = AutoConfig.from_pretrained("gpt2")
57
-model = IPEXModelForCausalLM.from_pretrained(
+- model = AutoModelForCausalLM.from_pretrained(
58
++ model = IPEXModelForCausalLM.from_pretrained(
59
"gpt2",
60
config=config,
61
torch_dtype=torch.bfloat16,
0 commit comments