fix(components): Upload the tuned adapter to Model Registry instead o…

…f model checkpoint from `preview.llm.rlhf_pipeline` PiperOrigin-RevId: 591117156
kubeflow · Dec 15, 2023 · e2e7322 · e2e7322
1 parent 75ffbf9
commit e2e7322
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/components/google-cloud/RELEASE.md b/components/google-cloud/RELEASE.md
@@ -1,4 +1,5 @@
 ## Upcoming release
+* Upload the tuned adapter to Model Registry instead of model checkpoint from `preview.llm.rlhf_pipeline`.
 
 ## Release 2.8.0
 * Release AutoSxS pipeline to preview.

diff --git a/components/google-cloud/google_cloud_pipeline_components/preview/llm/rlhf/component.py b/components/google-cloud/google_cloud_pipeline_components/preview/llm/rlhf/component.py
@@ -128,7 +128,7 @@ def rlhf_pipeline(
     )
 
   llm_model_handler = deployment_graph.pipeline(
-      output_adapter_path=rl_model_pipeline.outputs['output_model_path'],
+      output_adapter_path=rl_model_pipeline.outputs['output_adapter_path'],
       large_model_reference=large_model_reference,
       model_display_name=model_display_name,
       deploy_model=deploy_model,