Update README.md

README.md

@@ -41,4 +41,23 @@ ds = load_dataset("open-thoughts/OpenThoughts-114k", "default")
 from adapters import AutoAdapterModel
 
 model = AutoAdapterModel.from_pretrained("undefined")
-model.load_adapter("rebekah0302/Glo-Bus", set_active=True)
+model.load_adapter("rebekah0302/Glo-Bus", set_active=True)
+from transformers import TrainingArguments, Trainer
+
+training_args = TrainingArguments(
+    output_dir="./results",
+    evaluation_strategy="epoch",
+    per_device_train_batch_size=4,
+    per_device_eval_batch_size=4,
+    num_train_epochs=3,
+    save_total_limit=2,
+    logging_dir="./logs",
+)
+
+trainer = Trainer(
+    model=model,
+    args=training_args,
+    train_dataset=tokenized_datasets["train"],
+)
+
+trainer.train()