clibrain
/

mamba-2.8b-chat-no_robots

Text Generation

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

mrm8488 commited on Dec 12, 2023

Commit

09a0ce7

·

1 Parent(s): c2fce10

Update README.md

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

@@ -61,6 +61,7 @@ from mamba_ssm.models.mixer_seq_simple import MambaLMHeadModel
 CHAT_TEMPLATE_ID = "HuggingFaceH4/zephyr-7b-beta"
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 eos_token = "<|endoftext|>"
 tokenizer = AutoTokenizer.from_pretrained(model_name)

 CHAT_TEMPLATE_ID = "HuggingFaceH4/zephyr-7b-beta"
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model_name = "clibrain/mamba-2.8b-chat-no_robots"
 eos_token = "<|endoftext|>"
 tokenizer = AutoTokenizer.from_pretrained(model_name)