Fix CodeGemma model ID (#2873)

Co-authored-by: minwoopark <minwoopark@google.com>
This commit is contained in:
Minwoo Park
2024-04-09 15:23:17 +00:00
committed by GitHub
co-authored by minwoopark
parent 6980995572
commit fbb27557e8
@@ -187,7 +187,7 @@
" ! curl -X GET \"{signed_url}\" | tar -xzvf - -C ./codegemma/\n",
" ! gsutil -m cp -R ./codegemma/* {BUCKET_URI}\n",
"\n",
" model_path_prefix = BUCKET_URI\n",
" model_path_prefix = BUCKET_URI.strip(\"/\") + \"/codegemma\"\n",
"else:\n",
" model_path_prefix = \"google/\"\n",
"# The pre-built serving docker images with vLLM\n",