xiaohei/model_garden
1
0
Fork 0
mirror of https://github.com/GoogleCloudPlatform/vertex-ai-samples.git synced 2026-09-26 14:42:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: new model value (#2207)

Browse Source
This commit is contained in:
Andrew Ferlitsch
2023-09-18 16:24:08 +00:00
committed by GitHub
parent e4895da43b
commit e27db74595
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
notebooks/official/gen_ai/rlhf_tune_llm.ipynb
+2 -2
View File
@@ -103,7 +103,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",
@@ -620,7 +620,7 @@
" \"preference_dataset\": \"gs://vertex-ai/generative-ai/rlhf/text_small/summarize_from_feedback_tfds/comparisons/train/*.jsonl\",\n",
" \"prompt_dataset\": \"gs://vertex-ai/generative-ai/rlhf/text_small/reddit_tfds/train/*.jsonl\",\n",
" \"eval_dataset\": \"gs://vertex-ai/generative-ai/rlhf/text_small/reddit_tfds/val/*.jsonl\",\n",
" \"large_model_reference\": \"BISON\", # GA value will be text-bison@001\n",
" \"large_model_reference\": \"t5-xl\", # See table 1 for values\n",
" \"model_display_name\": \"my_rlhf_tutorial_model\", # Optional. If omitted, a default model_display_name will be created.\n",
" \"reward_model_train_steps\": 100, # Please remember to read \"A Note on choosing train_steps\" section.\n",
" \"reinforcement_learning_train_steps\": 100, # Please remember to read \"A Note on choosing train_steps\" section.\n",