Update README.md

README.md

@@ -1,7 +1,9 @@
 ---
-language: 
+language:
 - en
 license: mit
+datasets:
+- PatrickHaller/wiki-and-book-corpus-10M
 ---
 
 # An xLSTM Model
@@ -24,7 +26,7 @@ conda activate xlstm
 ```python
 from transformers import AutoModelForCasualLM, AutoTokenizer
 
-model_name_or_path = "PatrickHaller/xlstm_dummy"
+model_name_or_path = "PatrickHaller/xlstm_wikipedia_110M_10M"
 
 model = AutoModelForCasualLM.from_pretrained(model_name_or_path)
 tokenizer = AutoTokenizer.from_pretrained(model_name_or_path)
@@ -35,4 +37,4 @@ generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
 
 print(generated_text)
 
-```
+```