Update README.md

README.md

@@ -39,8 +39,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
 model_name = "mahing/historical-narrative-generator"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16)
+tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2.5-14B-Instruct-1M")
+model  = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto", torch_dtype=torch.bfloat16)
 
 event = 'The Magna Carta'
 prompt = f"Q: Craft a compelling first-person historical narrative that captures the significance of {event} and the essence of its era.