Compare commits

...
1 Commits
Author SHA1 Message Date
Andrew Ferlitsch 779f136b35 fix: new model value 2023-08-24 22:04:17 +00:00
@@ -99,7 +99,7 @@
"\n",
"| large_model_reference | supported accelerator_type | supported region |\n",
"|---|---|---|\n",
"| `BISON` | `TPU_V3`, `NVIDIA_TESLA_A100` | europe-west4, us-central1 |\n",
"| `text-bison@001` | `TPU_V3`, `NVIDIA_TESLA_A100` | europe-west4, us-central1 |\n",
"| `t5-small` | `TPU_V3`, `NVIDIA_TESLA_A100` | europe-west4, us-central1 |\n",
"| `t5-large` | `TPU_V3` | europe-west4 |\n",
"| `t5-xl` | `TPU_V3` | europe-west4 |\n",
@@ -538,7 +538,7 @@
" pipeline_root=PIPELINE_ROOT,\n",
" template_path=INFER_PIPELINE_PKG_PATH,\n",
" parameter_values={\n",
" \"large_model_reference\": \"BISON\", # see Table 1 for valid values.\n",
" \"large_model_reference\": \"t5-xl\", # see Table 1 for valid values.\n",
" \"model_checkpoint\": MODEL_CHECKPOINT,\n",
" \"prompt_dataset\": \"gs://vertex-ai/generative-ai/rlhf/text_small/reddit_tfds/val/*.jsonl\",\n",
" \"instruction\": \"Summarize in less than 50 words.\",\n",