fix

README.md

@@ -87,13 +87,13 @@ from mamba_gpt_pipeline import MambaGPTTextGenerationPipeline
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
 tokenizer = AutoTokenizer.from_pretrained(
-    "CobraMamba/mamba-gpt-3b",
+    "CobraMamba/mamba-gpt-3b-v2",
     use_fast=False,
     padding_side="left",
     trust_remote_code=False,
 )
 model = AutoModelForCausalLM.from_pretrained(
-    "CobraMamba/mamba-gpt-3b",
+    "CobraMamba/mamba-gpt-3b-v2",
     torch_dtype="auto",
     device_map={"": "cuda:0"},
     trust_remote_code=False,
@@ -119,7 +119,7 @@ You may also construct the pipeline from the loaded model and tokenizer yourself
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-model_name = "CobraMamba/mamba-gpt-3b"  # either local folder or huggingface model name
+model_name = "CobraMamba/mamba-gpt-3b-v2"  # either local folder or huggingface model name
 # Important: The prompt needs to be in the same format the model was trained with.
 # You can find an example prompt in the experiment logs.
 prompt = "<|prompt|>How are you?</s><|answer|>"