wenhuach commited on
Commit
f01205c
·
verified ·
1 Parent(s): 5b294fa

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
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(model_id, 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
  ```
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