Merge pull request #27 from rupaut98/fix-issue-24

Update Kaggle-Orpheus_(3B)-TTS.ipynb to fix issue #24

Author: shimmyshimmer

nb/Kaggle-Orpheus_(3B)-TTS.ipynb

@@ -527,6 +527,11 @@
     "from transformers import TrainingArguments,Trainer,DataCollatorForSeq2Seq\n",
     "from unsloth import is_bfloat16_supported\n",
     "\n",
+    "data_collator = DataCollatorForSeq2Seq(\n",
+    "    tokenizer=tokenizer,\n",
+    "    padding=True,\n",
+    ")\n",
+    "\n",
     "trainer = Trainer(\n",
     "    model = model,\n",
     "    train_dataset = dataset,\n",
@@ -547,8 +552,9 @@
     "        output_dir = \"outputs\",\n",
     "        report_to = \"none\", # Use this for WandB etc\n",
     "    ),\n",
+    "    data_collator=data_collator,\n",
     ")"
-   ]
+  ]
   },
   {
    "cell_type": "code",
@@ -4021,4 +4027,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 0
-}
+}