AntonV
/

mamba2-130m-hf

Inference Endpoints

Model card Files Files and versions Community

AntonV commited on 5 days ago

Commit

05e8773

•

1 Parent(s): 97d2a95

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -12,7 +12,7 @@ Not affiliated with both the original authors or hf.
 ## Usage
 ```python
-from transformers import AutoModelForCausalLM
 tokenizer = AutoTokenizer.from_pretrained("AntonV/mamba2-130m-hf")
 model = AutoModelForCausalLM.from_pretrained("AntonV/mamba2-130m-hf")

 ## Usage
 ```python
+from transformers import AutoTokenizer, AutoModelForCausalLM
 tokenizer = AutoTokenizer.from_pretrained("AntonV/mamba2-130m-hf")
 model = AutoModelForCausalLM.from_pretrained("AntonV/mamba2-130m-hf")