Update README.md

README.md

@@ -34,8 +34,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 device = "cuda"  # the device to load the model onto
 
 # Load the model and tokenizer
-model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2")
-tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2")
+model = AutoModelForCausalLM.from_pretrained("cognitivecomputations/DeepMixtral-8x7b-Instruct", trust_remote_code=True, device_map="auto")
+tokenizer = AutoTokenizer.from_pretrained("cognitivecomputations/DeepMixtral-8x7b-Instruct")
 
 # Define the conversation messages
 messages = [