Hugging Face's logo

state-spaces
/
mamba-1.4b-hf

Text Generation
Transformers
Safetensors
mamba
text-generation-inference
Model card Files Community
2
ArthurZ HF Staff commited on
Commit
cb6b717
·
verified ·
1 Parent(s): c3bd959

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -47,7 +47,7 @@ from datasets import load_dataset
47
  from trl import SFTTrainer
48
  from peft import LoraConfig
49
  from transformers import AutoTokenizer, AutoModelForCausalLM, TrainingArguments
50
- tokenizer = AutoTokenizer.from_pretrained("state-spaces/mamba-1.4b-hf", pad_token ="<s>")
51
  model = AutoModelForCausalLM.from_pretrained("state-spaces/mamba-1.4b-hf")
52
  dataset = load_dataset("Abirate/english_quotes", split="train")
53
  training_args = TrainingArguments(
 
47
  from trl import SFTTrainer
48
  from peft import LoraConfig
49
  from transformers import AutoTokenizer, AutoModelForCausalLM, TrainingArguments
50
+ tokenizer = AutoTokenizer.from_pretrained("state-spaces/mamba-1.4b-hf")
51
  model = AutoModelForCausalLM.from_pretrained("state-spaces/mamba-1.4b-hf")
52
  dataset = load_dataset("Abirate/english_quotes", split="train")
53
  training_args = TrainingArguments(