Update README.md

README.md

@@ -43,8 +43,8 @@ pip install auto-gptq
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
 quantized_model_dir = "Intel/phi-2-int4-inc"
-tokenizer = AutoTokenizer.from_pretrained(quantized_model_dir)
-model = AutoModelForCausalLM.from_pretrained(quantized_model_dir, device_map="auto", trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained(quantized_model_dir, revision="5973e3ad50beaefb937345d693639ce92ca836f9")
+model = AutoModelForCausalLM.from_pretrained(quantized_model_dir, device_map="auto", trust_remote_code=True, revision="5973e3ad50beaefb937345d693639ce92ca836f9")
 text = "There is a girl who likes adventure,"
 inputs = tokenizer(text, return_tensors="pt", return_attention_mask=False).to(model.device)
 outputs = model.generate(**inputs, max_new_tokens=50)