Update README.md

New pipeline usage

README.md

@@ -46,18 +46,20 @@ generate_text = pipeline(model="databricks/dolly-v2-3b", torch_dtype=torch.bfloa
 You can then use the pipeline to answer instructions:
 
 ```
-generate_text("Explain to me the difference between nuclear fission and fusion.")
+res = generate_text("Explain to me the difference between nuclear fission and fusion.")
+print(res[0]["generated_text"])
 ```
 
 Alternatively, if you prefer to not use `trust_remote_code=True` you can download [instruct_pipeline.py](https://huggingface.co/databricks/dolly-v2-3b/blob/main/instruct_pipeline.py),
 store it alongside your notebook, and construct the pipeline yourself from the loaded model and tokenizer:
 
 ```
+import torch
 from instruct_pipeline import InstructionTextGenerationPipeline
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
 tokenizer = AutoTokenizer.from_pretrained("databricks/dolly-v2-3b", padding_side="left")
-model = AutoModelForCausalLM.from_pretrained("databricks/dolly-v2-3b", device_map="auto")
+model = AutoModelForCausalLM.from_pretrained("databricks/dolly-v2-3b", device_map="auto", torch_dtype=torch.bfloat16)
 
 generate_text = InstructionTextGenerationPipeline(model=model, tokenizer=tokenizer)
 ```