From def3ac61ee07f9647117706526d45a25bddcecb3 Mon Sep 17 00:00:00 2001 From: Ryan Mullins Date: Tue, 9 Jul 2024 16:02:16 +0000 Subject: [PATCH] Fixing docstring for datasets_config in notebook cell --- site/en/gemma/docs/lit_gemma.ipynb | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/site/en/gemma/docs/lit_gemma.ipynb b/site/en/gemma/docs/lit_gemma.ipynb index 3b701fbb9..1d79c35d0 100644 --- a/site/en/gemma/docs/lit_gemma.ipynb +++ b/site/en/gemma/docs/lit_gemma.ipynb @@ -209,10 +209,10 @@ "# two required positional arguments:\n", "#\n", "# * `datasets_config`: A list of strings containing the dataset names and\n", - "# paths to load from, as \"dataset:path\", where path can be a URL, a local\n", - "# file path, or the name of a preset for the configured deep learning\n", - "# framework. Here, we use a special value, `sample_prompts`, to load the\n", - "# example prompts provided in the LIT distribution.\n", + "# paths to load from, as \"dataset:path\", where path can be a URL or a\n", + "# local file path. Here, we use a special value, `sample_prompts`, to load\n", + "# the example prompts provided in the LIT distribution; no other special\n", + "# values are supported.\n", "# * `models_config`: A list of strings containing the model names and paths to\n", "# load from, as \"model:path\", where path can be a URL, a local file path,\n", "# or the name of a preset for the configured deep learning framework.\n", @@ -225,7 +225,8 @@ "# * `dl_framework`: Must be one of \"kerasnlp\" or \"transformers\".\n", "# * `dl_runtime`: Must be one of \"tensorflow\" or \"torch\".\n", "#\n", - "# Changing the `dl_framework` value may affect how you access the Gemma weights.\n", + "# Changing the `dl_framework` value will affect the authentication method used\n", + "# to access Gemma model weights.\n", "\n", "lit_widget = lit_pdbnb.make_notebook_widget(\n", " ['sample_prompts'],\n",