correct example snippet

apache · Sep 19, 2023 · e0fca8a · e0fca8a
1 parent e2397fb
commit e0fca8a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sdks/python/apache_beam/ml/inference/huggingface_inference.py b/sdks/python/apache_beam/ml/inference/huggingface_inference.py
@@ -603,7 +603,7 @@ def __init__(
 
         Example Usage::
           model_handler = HuggingFacePipelineModelHandler(
-            model_uri="bert-base-uncased", model="meta-llama/Llama-2-7b-hf",
+            task="text-generation", model="meta-llama/Llama-2-7b-hf",
             load_pipeline_args={'model_kwargs':{'quantization_map':config}})
 
       inference_fn: the inference function to use during RunInference.