Fix import in notebook

finetune.ipynb

@@ -28,8 +28,8 @@
     "# This should point to your Truefoundry platform endpoint\n",
     "TRUEFOUNDRY_HOST = os.getenv(\"TFY_HOST\", \"https://<your-org>.truefoundry.cloud\")\n",
     "\n",
-    "from truefoundry import ml as mlfoundry\n",
-    "mlfoundry.login(TRUEFOUNDRY_HOST)"
+    "import truefoundry\n",
+    "truefoundry.login(TRUEFOUNDRY_HOST)"
    ]
   },
   {
@@ -265,7 +265,7 @@
     "micro_batch_size = launch_parameters.batch_size\n",
     "\n",
     "# Learning rate\n",
-    "learning_rate = 0.00003\n",
+    "learning_rate = 0.00009\n",
     "\n",
     "# How many epochs to run training for\n",
     "num_epochs = 10\n",
@@ -423,6 +423,13 @@
    "source": [
     "!{COMMAND}"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": []
   }
  ],
  "metadata": {

sample_run.sh

@@ -43,7 +43,7 @@ config-base.yaml \
 --num_epochs 5 \
 --gradient_accumulation_steps 4 \
 --gradient_checkpointing unsloth \
---learning_rate 0.00001 \
+--learning_rate 0.00009 \
 --output_dir ./outputs \
 --train_on_inputs False \
 --logging_steps 1 \