Add tag

Author: windmaple

docs/source/JAX_for_LLM_pretraining.ipynb

@@ -1924,7 +1924,10 @@
      "height": 799
     },
     "id": "-uTeXvXaLCZb",
-    "outputId": "aa52154f-9583-46d3-896a-05e28de603de"
+    "outputId": "aa52154f-9583-46d3-896a-05e28de603de",
+    "tags": [
+     "nbval-ignore-output"
+    ]
    },
    "outputs": [
     {
@@ -1996,10 +1999,7 @@
   {
    "cell_type": "markdown",
    "metadata": {
-    "id": "3813cbf2",
-    "tags": [
-     "nbval-ignore-output"
-    ]
+    "id": "3813cbf2"
    },
    "source": [
     "## Xprof: profiling for hyperparameter tuning\n",

docs/source/JAX_for_LLM_pretraining.md

@@ -850,13 +850,14 @@ colab:
   height: 799
 id: -uTeXvXaLCZb
 outputId: aa52154f-9583-46d3-896a-05e28de603de
+tags: [nbval-ignore-output]
 ---
 # Generate text with LoRA-finetuned model
 print("Generating text after LoRA finetuning:\n\n")
 lora_model.generate_text(maxlen, start_tokens)
 ```
 
-+++ {"id": "3813cbf2", "tags": ["nbval-ignore-output"]}
++++ {"id": "3813cbf2"}
 
 ## Xprof: profiling for hyperparameter tuning