Update README.md

README.md

@@ -49,7 +49,11 @@ tokenizer = AutoTokenizer.from_pretrained(quantized_model_dir)
 model = AutoModelForCausalLM.from_pretrained(quantized_model_dir,
                                              device_map="auto"
                                              )
+
+##import habana_frameworks.torch.core as htcore ## uncommnet it for HPU
+##import habana_frameworks.torch.hpu as hthpu ## uncommnet it for HPU
 ##model = model.to(torch.bfloat16).to("hpu") ## uncommnet it for HPU
+
 tokenizer = AutoTokenizer.from_pretrained(quantized_model_dir)
 text = "There is a girl who likes adventure,"
 inputs = tokenizer(text, return_tensors="pt").to(model.device)