Update README.md

README.md

@@ -12,14 +12,11 @@ This model outperforms an fp16 LLama-2-13B (perplexity 4.13 vs. 4.63) for a comp
 
 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-70b-hf-2bit_g16_s128-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>