ArthurZ
/

mamba-130m

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

devingulliver commited on Apr 11

Commit

2f3fa58

•

1 Parent(s): 80bb8cf

Update README.md

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']
+```