Adding comparison section

notebooks/sft.json

@@ -62,7 +62,19 @@
         },
         {
             "cell_type": "markdown",
-            "source": "## 4. Push model to hub (Optional)"
+            "source": "## 4. Compare model output vs original"
+        },
+        {
+            "cell_type": "code",
+            "source": "from transformers import pipeline, AutoTokenizer\n\nmy_model = trainer.model\ntokenizer = AutoTokenizer.from_pretrained(model_name)\n\nmy_pipe = pipeline('text-generation', model=my_model, tokenizer=tokenizer)\noriginal_pipe = pipeline('text-generation', model=model_name)"
+        },
+        {
+            "cell_type": "code",
+            "source": "print(original_pipe('Hello'))\nprint(my_pipe('Hello'))"
+        },
+        {
+            "cell_type": "markdown",
+            "source": "## 5. Push model to hub (Optional)"
         },
         {
             "cell_type": "code",