Update README.md

README.md

@@ -60,6 +60,11 @@ model = AutoModelForCausalLM.from_pretrained(
     torch_dtype='auto',
     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
+
 prompt = "There is a girl who likes adventure,"
 messages = [
     {"role": "system", "content": "You are a helpful assistant."},
@@ -86,11 +91,9 @@ output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_
 response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
 print(response)
 
-##prompt = "请介绍一下阿里巴巴公司"
-##阿里巴巴集团是中国最大的电子商务公司之一，由马云等18人于1999年在杭州创立。阿里巴巴集团旗下拥有多个知名的电子商务平台，包括�
 
-##prompt = "9.8大还是9.11大"
-##9.11大
+##prompt = "Which one is bigger, 9.11 or 9.8"
+##9.11 is bigger than 9.8.
 
 ##prompt = "Once upon a time,"
 ##it seems like we're about to start a classic fairy tale. Would you like to continue the story, or would you like me to take over and spin a yarn for you?