Update README.md
Browse files
README.md
CHANGED
|
@@ -50,11 +50,12 @@ Install [AutoGPTQ](https://github.com/AutoGPTQ/AutoGPTQ) from source first
|
|
| 50 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 51 |
quantized_model_dir = "Intel/phi-2-int4-inc"
|
| 52 |
tokenizer = AutoTokenizer.from_pretrained(quantized_model_dir)
|
| 53 |
-
model = AutoModelForCausalLM.from_pretrained(
|
| 54 |
text = "There is a girl who likes adventure,"
|
| 55 |
inputs = tokenizer(text, return_tensors="pt", return_attention_mask=False).to(model.device)
|
| 56 |
outputs = model.generate(**inputs, max_new_tokens=50)
|
| 57 |
text = tokenizer.batch_decode(outputs)[0]
|
|
|
|
| 58 |
```
|
| 59 |
|
| 60 |
|
|
|
|
| 50 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 51 |
quantized_model_dir = "Intel/phi-2-int4-inc"
|
| 52 |
tokenizer = AutoTokenizer.from_pretrained(quantized_model_dir)
|
| 53 |
+
model = AutoModelForCausalLM.from_pretrained(quantized_model_dir, device_map="auto", trust_remote_code=True)
|
| 54 |
text = "There is a girl who likes adventure,"
|
| 55 |
inputs = tokenizer(text, return_tensors="pt", return_attention_mask=False).to(model.device)
|
| 56 |
outputs = model.generate(**inputs, max_new_tokens=50)
|
| 57 |
text = tokenizer.batch_decode(outputs)[0]
|
| 58 |
+
print(text)
|
| 59 |
```
|
| 60 |
|
| 61 |
|