CobraMamba
/

mamba-gpt-3b-v2

Text Generation

large language model

text-generation-inference

Model card Files Files and versions Community

chiliu commited on Aug 2, 2023

Commit

edf5984

•

1 Parent(s): db69dce

update

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -49,7 +49,7 @@ import torch
 from transformers import pipeline
 generate_text = pipeline(
-    model="CobraMamba/mamba-gpt-3b",
     torch_dtype="auto",
     trust_remote_code=True,
     use_fast=False,

 from transformers import pipeline
 generate_text = pipeline(
+    model="CobraMamba/mamba-gpt-3b-v2",
     torch_dtype="auto",
     trust_remote_code=True,
     use_fast=False,