diff --git a/tutorials/llm/llama-3/nemo2-sft-peft/nemo2-sft.ipynb b/tutorials/llm/llama-3/nemo2-sft-peft/nemo2-sft.ipynb index ccd645e047c8..71dd1c264698 100644 --- a/tutorials/llm/llama-3/nemo2-sft-peft/nemo2-sft.ipynb +++ b/tutorials/llm/llama-3/nemo2-sft-peft/nemo2-sft.ipynb @@ -106,15 +106,9 @@ }, "outputs": [], "source": [ - "from datasets import load_dataset\n", "from typing import List, Optional\n", - "from nemo.collections.common.tokenizers import TokenizerSpec\n", "from nemo.lightning.io.mixin import IOMixin\n", "from nemo.collections.llm.gpt.data.fine_tuning import FineTuningDataModule\n", - "import json\n", - "from nemo.utils import logging\n", - "import numpy as np\n", - "import shutil\n", "\n", "class DollyDataModule(FineTuningDataModule, IOMixin):\n", " def __init__(\n",