Update README.md

README.md

@@ -14,7 +14,8 @@ It was fine-tuned on several thousand astronomy abstracts collected on Arxiv.
 ```python
 from transformers import AutoModelForCausalLM
 
-online_model = AutoModelForCausalLM.from_pretrained("charlieoneill/falcon-7b-abstracts-2190", trust_remote_code=True)
+online_model = AutoModelForCausalLM.from_pretrained("charlieoneill/falcon-7b-abstracts-2190", torch_dtype=torch.bfloat16,
+                                                    trust_remote_code=True)
 
 tokenizer = AutoTokenizer.from_pretrained("tiiuae/falcon-7b")
 pipeline = transformers.pipeline(