Skip to content

Commit 09f5b6b

Browse files
committed
add hugging face info
1 parent 9cda5a3 commit 09f5b6b

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

sdks/python/apache_beam/yaml/yaml_ml.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -492,6 +492,7 @@ def fn(x: PredictionResult):
492492
handler_config parameters, see their corresponding doc pages:
493493
494494
- [VertexAIModelHandlerJSON](https://beam.apache.org/releases/pydoc/current/apache_beam.yaml.yaml_ml.VertexAIModelHandlerJSONProvider) # pylint: disable=line-too-long
495+
- [HuggingFacePipelineProvider](https://beam.apache.org/releases/pydoc/current/apache_beam.yaml.yaml_ml.HuggingFacePipelineProvider) # pylint: disable=line-too-long
495496
inference_tag: The tag to use for the returned inference. Default is
496497
'inference'.
497498
inference_args: Extra arguments for models whose inference call requires

0 commit comments

Comments
 (0)