small changes to torchtune notebook + reran Comet example project

Author: anmorgan24

integrations/model-training/torchtune/notebooks/Comet_and_torchtune.ipynb

@@ -36,7 +36,7 @@
    "source": [
     "\n",
     "<a href=\"https://www.comet.com/examples/comet-examples-torchtune-mistral7b?utm_medium=colab&utm_source=comet-examples&utm_campaign=torchtune\" >\n",
-    "  <img src=\"https://raw.githubusercontent.com/anmorgan24/Comet_Projects/main/Project%20Images/torchtune_and_comet.png\">\n",
+    "  <img src=\"https://raw.githubusercontent.com/anmorgan24/Comet_Projects/main/Project%20Images/comet_and_torchtune.png\">\n",
     "</a>\n",
     "\n",
     "\n",
@@ -79,7 +79,7 @@
    },
    "outputs": [],
    "source": [
-    "%pip install \"./torchtune[dev]\" comet_ml"
+    "%pip install \"./torchtune[dev]\" comet_ml torchao"
    ]
   },
   {
@@ -214,7 +214,7 @@
     "\n",
     "```yaml\n",
     "metric_logger:\n",
-    "  _component_: torchtune.utils.metric_logging.CometLogger\n",
+    "  _component_: torchtune.training.metric_logging.CometLogger\n",
     "```"
    ]
   },
@@ -237,7 +237,7 @@
     "\n",
     "# Dataset\n",
     "dataset:\n",
-    "  _component_: torchtune.datasets.alpaca_cleaned_dataset\n",
+    "  _component_: torchtune.datasets.alpaca_dataset\n",
     "  train_on_input: True\n",
     "seed: null\n",
     "shuffle: True\n",