Update README.md

README.md

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