Update README.md

README.md

@@ -1,4 +1,4 @@
-Model Card for Extended-Mind-MPT-7b-chat
+Model Card for Extended-Mind-Mpt-7b-chat
 
 * Github: https://github.com/normal-computing/extended-mind-transformers/
 * ArXiv: https://arxiv.org/abs/2406.02332
@@ -27,10 +27,10 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 
 ag_wiki_entry = """Alexander Grothendieck (/ˈɡroʊtəndiːk/; German pronunciation: [ˌalɛˈksandɐ ˈɡʁoːtn̩ˌdiːk] (listen); French: [ɡʁɔtɛndik]; 28 March 1928 – 13 November 2014) was a stateless (and then, since 1971, French) mathematician who became the leading figure in the creation of modern algebraic geometry.[7][8] His research extended the scope of the field and added elements of commutative algebra, homological algebra, sheaf theory, and category theory to its foundations, while his so-called "relative" perspective led to revolutionary advances in many areas of pure mathematics.[7][9] He is considered by many to be the greatest mathematician of the twentieth century.[10][11]"""
 
-tokenizer_hf = AutoTokenizer.from_pretrained("normalcomputing/extended-mind-llama-2-7b")
+tokenizer_hf = AutoTokenizer.from_pretrained("normalcomputing/extended-mind-mpt-7b-chat")
 memories = tokenizer_hf(ag_wiki_entry).input_ids
 
-model_hf = AutoModelForCausalLM.from_pretrained("normalcomputing/extended-mind-llama-2-7b", external_memories=memories, trust_remote_code=True)
+model_hf = AutoModelForCausalLM.from_pretrained("normalcomputing/extended-mind-mpt-7b-chat", external_memories=memories, trust_remote_code=True)
 ```
 After this, you can generate text with the model as usual. The model will automatically use the memories during generation. You can update any config parameters (we set `topk` below) by passing new values to the `model.generate()` method.