Skip to content

Commit

Permalink
Disable kfp for hap. It will be addressed in a follow-up PR
Browse files Browse the repository at this point in the history
Signed-off-by: Maroun Touma <touma@us.ibm.com>
  • Loading branch information
touma-I committed Oct 24, 2024
1 parent 10dcb89 commit e7625ef
Show file tree
Hide file tree
Showing 3 changed files with 44 additions and 0 deletions.
File renamed without changes.
File renamed without changes.
44 changes: 44 additions & 0 deletions transforms/universal/hap/kfp_ray.disable/pipeline_definitions.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,44 @@
pipeline_parameters:
name: "hap"
description: "Pipeline for hap task"
script_name: "hap_transform_ray.py"
prefix: ""
multi_s3: False
compute_func_name: ""
compute_func_import: ""
component_spec_path: ""

pipeline_common_input_parameters_values:
kfp_base_image: "quay.io/dataprep1/data-prep-kit/kfp-data-processing:latest"
transform_image: "quay.io/dataprep1/data-prep-kit/hap-ray:latest"
s3_access_secret: "s3-secret"
image_pull_secret: ""
input_folder: "test/hap/input/"
output_folder: "test/hap/output/"

pipeline_transform_input_parameters:
pipeline_arguments:
- name: "model_name_or_path"
type: "str"
value: "ibm-granite/granite-guardian-hap-38m"
description: "# HAP model path"
- name: "annotation_column"
type: "str"
value: "hap_score"
description: "# hap score for each document"
- name: "doc_text_column"
type: "str"
value: "contents"
description: "# The column name that contains the document text"
- name: "inference_engine"
type: "str"
value: "CPU"
description: "# inference engine used"
- name: max_length
type: "int"
value: 512
description: "# inference engine used"
- name: "batch_size"
type: "int"
value: 128
description: "# batch size"

0 comments on commit e7625ef

Please sign in to comment.