Create README.md

README.md

@@ -0,0 +1,42 @@
+---
+license: apache-2.0
+language:
+- en
+tags:
+- mamba-hf
+---
+
+# Mamba-3B
+
+<img src="https://th.bing.com/th/id/OIG.Jp5dA01tOAFcwSp544nv?pid=ImgGn" width="300" height="300" alt="mamba-hf">
+
+Mamba Models with hf_integration.
+[**mamba-hf** Github Repo](https://github.com/LegallyCoder/mamba-hf)
+
+# Usage:
+
+```python
+from transformers import AutoModelForCausalLM , AutoTokenizer
+
+model = AutoModelForCausalLM.from_pretrained('Q-bert/Mamba-3B', trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained('Q-bert/Mamba-3B')
+
+text = "Hi"
+
+input_ids = tokenizer.encode(text, return_tensors="pt")
+
+output = model.generate(input_ids, max_length=20, num_beams=5, no_repeat_ngram_size=2)
+
+generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
+
+print(generated_text)
+
+```
+> Hi, I'm looking for a new job. I've been working at a company for about a year now.
+
+# Credits:
+
+https://huggingface.co/state-spaces
+
+Special thanks to Albert Gu and Tri Dao for their articles. (https://arxiv.org/abs/2312.00752)
+