Update README.md
Browse files
README.md
CHANGED
@@ -26,8 +26,7 @@ from auto_round.auto_quantizer import AutoHfQuantizer
|
|
26 |
from transformers import AutoModelForCausalLM,AutoTokenizer
|
27 |
quantized_model_dir = "Intel/Qwen2-7B-int4-inc"
|
28 |
tokenizer = AutoTokenizer.from_pretrained(quantized_model_dir)
|
29 |
-
model = AutoModelForCausalLM.from_pretrained(
|
30 |
-
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
31 |
text = "下面我来介绍一下阿里巴巴公司,"
|
32 |
text = "88+99等于多少?"
|
33 |
text = "Once upon a time,"
|
|
|
26 |
from transformers import AutoModelForCausalLM,AutoTokenizer
|
27 |
quantized_model_dir = "Intel/Qwen2-7B-int4-inc"
|
28 |
tokenizer = AutoTokenizer.from_pretrained(quantized_model_dir)
|
29 |
+
model = AutoModelForCausalLM.from_pretrained(quantized_model_dir, device_map="auto")
|
|
|
30 |
text = "下面我来介绍一下阿里巴巴公司,"
|
31 |
text = "88+99等于多少?"
|
32 |
text = "Once upon a time,"
|