Update README.md

README.md

@@ -36,11 +36,16 @@ will report soon
 `garage-bAInd/Platypus2-70B` was instruction fine-tuned using gptq on 2 L40 48GB. 
 
 ## How to Get Started with the Model
+First install auto_gptq with 
+```shell
+pip install auto_gptq
+```
 
 Use the code sample provided in the original post to interact with the model.
 ```python
-from transformers import AutoModelForCausalLM, AutoTokenizer
-
+from transformers import AutoTokenizer
+from auto_gptq import AutoGPTQForCausalLM
+ 
 model_id = "malhajar/Platypus2-70B-instruct-4bit-gptq"
 model = AutoGPTQForCausalLM.from_quantized(model_id,inject_fused_attention=False,
         use_safetensors=True,