Skip to content

Commit

Permalink
Fixing docstring for datasets_config in notebook cell
Browse files Browse the repository at this point in the history
  • Loading branch information
RyanMullins committed Jul 9, 2024
1 parent 37f346a commit def3ac6
Showing 1 changed file with 6 additions and 5 deletions.
11 changes: 6 additions & 5 deletions site/en/gemma/docs/lit_gemma.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand All @@ -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",
Expand Down

0 comments on commit def3ac6

Please sign in to comment.