Update README.md

README.md

@@ -12,14 +12,11 @@ This model matches the quality of AWQ (same perplexity) but didn't use any data
 
 To run the model, install the HQQ library from https://github.com/mobiusml/hqq and use it as follows:
 ``` Python
-from hqq.models.llama_hf import LlamaHQQ
-import transformers
-
 model_id = 'mobiuslabsgmbh/Llama-2-13b-hf-4bit_g64-HQQ'
-#Load the tokenizer
-tokenizer = transformers.AutoTokenizer.from_pretrained(model_id)
-#Load the model
-model = LlamaHQQ.from_quantized(model_id)
+
+from hqq.engine.hf import HQQModelForCausalLM, AutoTokenizer
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+model     = HQQModelForCausalLM.from_quantized(model_id)
 ```
 
 *Limitations*: <br>