ArthurZ
/

mamba-130m

Text Generation

text-generation-inference

Model card Files Files and versions

Update README.md

#1

by devingulliver - opened Apr 11, 2024

base: refs/heads/main

←

from: refs/pr/1

Discussion Files changed

Files changed (1) hide show

README.md +2 -3

README.md CHANGED Viewed

@@ -1,6 +1,6 @@
 ---
 library_name: transformers
-tags: []
 ---
 ```python
 >>> from transformers import MambaConfig, MambaForCausalLM, AutoTokenizer
@@ -16,5 +16,4 @@ tags: []
 >>> out = model.generate(input_ids, max_new_tokens=10)
 >>> print(tokenizer.batch_decode(out))
 ["<|endoftext|>Hey how are you doing?\n\nI'm a newbie to the game", 'Explain how soy sauce is made.\n\n1. Add the soy sauce to']
-```

 ---
 library_name: transformers
+license: apache-2.0
 ---
 ```python
 >>> from transformers import MambaConfig, MambaForCausalLM, AutoTokenizer
 >>> out = model.generate(input_ids, max_new_tokens=10)
 >>> print(tokenizer.batch_decode(out))
 ["<|endoftext|>Hey how are you doing?\n\nI'm a newbie to the game", 'Explain how soy sauce is made.\n\n1. Add the soy sauce to']
+```