Skip to content

Commit c967d9f

Browse files
author
Googler
committed
feat(components): use rlhf_preprocessor to replace the current value_exists call in rlhf
PiperOrigin-RevId: 627630447
1 parent c297119 commit c967d9f

File tree

2 files changed

+4
-5
lines changed

2 files changed

+4
-5
lines changed

components/google-cloud/google_cloud_pipeline_components/_implementation/llm/generated/refined_image_versions.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,4 +17,4 @@
1717
DO NOT EDIT - This file is generated, manual changes will be overridden.
1818
"""
1919

20-
IMAGE_TAG = '20240417_0507_RC00'
20+
IMAGE_TAG = '20240423_1336_RC00'

components/google-cloud/google_cloud_pipeline_components/preview/llm/rlhf/component.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -191,11 +191,10 @@ def rlhf_pipeline(
191191
encryption_spec_key_name=encryption_spec_key_name,
192192
).set_display_name('Reinforcement Learning')
193193

194-
has_inference_dataset = function_based.value_exists(
195-
value=eval_dataset
196-
).set_display_name('Resolve Inference Dataset')
194+
has_inference_dataset = preprocess_metadata.outputs['has_inference_dataset']
195+
197196
with kfp.dsl.Condition(
198-
has_inference_dataset.output == True, # pylint: disable=singleton-comparison
197+
has_inference_dataset == True, # pylint: disable=singleton-comparison
199198
name='Perform Inference',
200199
):
201200
has_model_checkpoint = function_based.value_exists(

0 commit comments

Comments
 (0)