state-spaces
/

mamba-370m-hf

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

ArthurZ HF staff commited on Mar 6

Commit

e96042a

•

1 Parent(s): 52557a5

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -46,7 +46,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-370m-hf", pad_token ="<s>")
 model = AutoModelForCausalLM.from_pretrained("state-spaces/mamba-370m-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-370m-hf")
 model = AutoModelForCausalLM.from_pretrained("state-spaces/mamba-370m-hf")
 dataset = load_dataset("Abirate/english_quotes", split="train")
 training_args = TrainingArguments(