state-spaces
/

mamba-790m-hf

Text Generation

text-generation-inference

Model card Files Files and versions

ArthurZ HF Staff commited on Mar 6, 2024

Commit

7e14e7a

·

verified ·

1 Parent(s): d285eea

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