Update README.md

README.md

@@ -1,5 +1,4 @@
-Small dummy LLama2-type Model useable for Unit/Integration tests. 
-Ensure that model input ids are < 100, see code below.
+Small dummy LLama2-type Model useable for Unit/Integration tests.
 
 
 ```python
@@ -14,13 +13,11 @@ config.intermediate_size = 24
 config.num_attention_heads = 2
 config.num_hidden_layers = 2
 config.num_key_value_heads = 2
-config.vocab_size = 100
 
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
-
 model = AutoModelForCausalLM.from_config(config)
-print(model.num_parameters())  # 5340
+print(model.num_parameters())  # 770_940
 
 model.push_to_hub(repo_name, private=False)
 tokenizer.push_to_hub(repo_name, private=False)