Update README.md
Browse files
README.md
CHANGED
|
@@ -5,36 +5,11 @@ datasets:
|
|
| 5 |
---
|
| 6 |
## Model Details
|
| 7 |
|
| 8 |
-
This model is an int4 model with group_size128 and sym quantization of [microsoft/phi-2](https://huggingface.co/microsoft/phi-2) generated by [intel/auto-round](https://github.com/intel/auto-round).
|
| 9 |
|
| 10 |
|
| 11 |
|
| 12 |
### How To Use
|
| 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 |
-
## revision="5973e3a" ##AutoGPTQ format
|
| 25 |
-
)
|
| 26 |
-
tokenizer = AutoTokenizer.from_pretrained(quantized_model_dir, use_fast=True)
|
| 27 |
-
print(tokenizer.decode(model.generate(**tokenizer("There is a girl who likes adventure,", return_tensors="pt").to(model.device),max_new_tokens=50)[0]))
|
| 28 |
-
"""
|
| 29 |
-
There is a girl who likes adventure,
|
| 30 |
-
She loves to explore and to venture.
|
| 31 |
-
She travels to faraway lands,
|
| 32 |
-
And meets people from different lands.
|
| 33 |
-
She learns new languages and cultures,
|
| 34 |
-
And makes friends with all kinds of people.
|
| 35 |
-
She is curious and brave and
|
| 36 |
-
"""
|
| 37 |
-
```
|
| 38 |
|
| 39 |
|
| 40 |
### INT4 Inference
|
|
|
|
| 5 |
---
|
| 6 |
## Model Details
|
| 7 |
|
| 8 |
+
This model is an int4 model with group_size128 and sym quantization of [microsoft/phi-2](https://huggingface.co/microsoft/phi-2) generated by [intel/auto-round](https://github.com/intel/auto-round). If you need AutoGPTQ format, please load the model with revision 5973e3a
|
| 9 |
|
| 10 |
|
| 11 |
|
| 12 |
### How To Use
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 13 |
|
| 14 |
|
| 15 |
### INT4 Inference
|