Jo1uck
/

mamba-11b-back

Text Generation

text-generation-inference

Model card Files Files and versions

Jo1uck commited on 28 days ago

Commit

b85e856

·

verified ·

1 Parent(s): 3a485f0

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -47,8 +47,8 @@ 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")
-model = AutoModelForCausalLM.from_pretrained("state-spaces/mamba-2.8b-hf")
 dataset = load_dataset("Abirate/english_quotes", split="train")
 training_args = TrainingArguments(
     output_dir="./results",

 from trl import SFTTrainer
 from peft import LoraConfig
 from transformers import AutoTokenizer, AutoModelForCausalLM, TrainingArguments
+tokenizer = AutoTokenizer.from_pretrained("Jo1uck/mamba-11b-back")
+model = AutoModelForCausalLM.from_pretrained("Jo1uck/mamba-11b-back")
 dataset = load_dataset("Abirate/english_quotes", split="train")
 training_args = TrainingArguments(
     output_dir="./results",