Skip to content

Commit

Permalink
fix(components): add check and add log to call out the fallback to th…
Browse files Browse the repository at this point in the history
…e default model checkpoint and remove the model checkpoint check condition in RLHF GCPC

Signed-off-by: Googler <nobody@google.com>
PiperOrigin-RevId: 642671232
  • Loading branch information
Googler committed Jun 12, 2024
1 parent de346d5 commit 92c3178
Showing 1 changed file with 3 additions and 6 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -206,12 +206,9 @@ def rlhf_pipeline(
has_inference_dataset == True, # pylint: disable=singleton-comparison
name='Perform Inference',
):
has_model_checkpoint = function_based.value_exists(
value=rl_model_pipeline.outputs['output_model_path']
).set_display_name('Resolve Model Checkpoint')
with kfp.dsl.Condition(
has_model_checkpoint.output == True, # pylint: disable=singleton-comparison
name='Test Model Checkpoint Exists',
with kfp.dsl.If(
rl_model_pipeline.outputs['output_model_path'] != '',
name='CheckModel Checkpoint Exists',
):
component.infer_pipeline(
project=project,
Expand Down

0 comments on commit 92c3178

Please sign in to comment.