yintongl commited on
Commit
f7c0764
·
verified ·
1 Parent(s): b275332

Update README.md

Browse files

add itrex inference eg

Files changed (1) hide show
  1. README.md +25 -0
README.md CHANGED
@@ -10,6 +10,31 @@ This model is an int4 model with group_size128 and sym quantization of [microsof
10
 
11
 
12
  ### Use the model
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
13
 
14
  ### INT4 Inference with AutoGPTQ
15
 
 
10
 
11
 
12
  ### Use the model
13
+ ### INT4 Inference with ITREX on CPU
14
+ Install the latest [intel-extension-for-transformers](
15
+ https://github.com/intel/intel-extension-for-transformers)
16
+ ```python
17
+ from intel_extension_for_transformers.transformers import AutoModelForCausalLM
18
+ from transformers import AutoTokenizer
19
+ quantized_model_dir = "Intel/phi-2-int4-inc"
20
+ model = AutoModelForCausalLM.from_pretrained(quantized_model_dir,
21
+ device_map="auto",
22
+ trust_remote_code=False,
23
+ use_neural_speed=False,
24
+ )
25
+ tokenizer = AutoTokenizer.from_pretrained(quantized_model_dir, use_fast=True)
26
+ print(tokenizer.decode(model.generate(**tokenizer("There is a girl who likes adventure,", return_tensors="pt").to(model.device),max_new_tokens=50)[0]))
27
+ """
28
+ There is a girl who likes adventure,
29
+ She loves to explore and to venture.
30
+ She travels to faraway lands,
31
+ And meets people from different lands.
32
+ She learns new languages and cultures,
33
+ And makes friends with all kinds of people.
34
+ She is curious and brave and
35
+ """
36
+ ```
37
+
38
 
39
  ### INT4 Inference with AutoGPTQ
40