state-spaces
/

mamba-2.8b-hf

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

ArthurZ HF staff commited on Mar 6

Commit

2b8a446

•

1 Parent(s): 8b6aa41

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -47,7 +47,7 @@ from datasets import load_dataset
 from trl import SFTTrainer
 from peft import LoraConfig
 from transformers import AutoTokenizer, AutoModelForCausalLM, TrainingArguments
-tokenizer = AutoTokenizer.from_pretrained("state-spaces/mamba-2.8b-hf", pad_token ="<s>")
 model = AutoModelForCausalLM.from_pretrained("state-spaces/mamba-2.8b-hf")
 dataset = load_dataset("Abirate/english_quotes", split="train")
 training_args = TrainingArguments(

 from trl import SFTTrainer
 from peft import LoraConfig
 from transformers import AutoTokenizer, AutoModelForCausalLM, TrainingArguments
+tokenizer = AutoTokenizer.from_pretrained("state-spaces/mamba-2.8b-hf")
 model = AutoModelForCausalLM.from_pretrained("state-spaces/mamba-2.8b-hf")
 dataset = load_dataset("Abirate/english_quotes", split="train")
 training_args = TrainingArguments(