Use lighter Colab defaults for news summarization notebook

eval/news_summarization/notebook.ipynb

@@ -13,7 +13,7 @@
         "This notebook pulls the benchmark code directly from the Hugging Face repo so it stays in sync with the HF page.\n",
         "\n",
         "Recommended first model:\n",
-        "- `Qwen/Qwen2.5-7B-Instruct`\n"
+        "- `Qwen/Qwen2.5-3B-Instruct`\n"
       ]
     },
     {
@@ -77,11 +77,12 @@
       },
       "outputs": [],
       "source": [
-        "MODEL = \"Qwen/Qwen2.5-7B-Instruct\"\n",
-        "LIMIT = 50\n",
+        "MODEL = \"Qwen/Qwen2.5-3B-Instruct\"\n",
+        "LIMIT = 10\n",
         "PROMPT_STYLE = \"simple\"\n",
         "DTYPE = \"bfloat16\"\n",
-        "TRUST_REMOTE_CODE = True"
+        "TRUST_REMOTE_CODE = False\n",
+        "DISABLE_BERTSCORE = True"
       ]
     },
     {
@@ -93,12 +94,14 @@
       "outputs": [],
       "source": [
         "trust_flag = \"--trust-remote-code\" if TRUST_REMOTE_CODE else \"\"\n",
+        "bertscore_flag = \"--disable-bertscore\" if DISABLE_BERTSCORE else \"\"\n",
         "!python eval/news_summarization/run_hf_transformers.py \\\n",
         "  --model {MODEL} \\\n",
         "  --limit {LIMIT} \\\n",
         "  --prompt-style {PROMPT_STYLE} \\\n",
         "  --dtype {DTYPE} \\\n",
         "  --verbose \\\n",
+        "  {bertscore_flag} \\\n",
         "  {trust_flag}"
       ]
     },
@@ -111,12 +114,14 @@
       "outputs": [],
       "source": [
         "trust_flag = \"--trust-remote-code\" if TRUST_REMOTE_CODE else \"\"\n",
+        "bertscore_flag = \"--disable-bertscore\" if DISABLE_BERTSCORE else \"\"\n",
         "!python eval/news_summarization/run_hf_transformers.py \\\n",
         "  --model {MODEL} \\\n",
         "  --limit {LIMIT} \\\n",
         "  --prompt-style {PROMPT_STYLE} \\\n",
         "  --dtype {DTYPE} \\\n",
         "  --resume \\\n",
+        "  {bertscore_flag} \\\n",
         "  {trust_flag}"
       ]
     },