backend

String

hugging_face
spacy

Backend to use for Named Entity Recognition.
Required.

model

String

Name of the model or a path to the model on the local disk. Dependent on the backend.
Required.

pipeline_kwargs

Dictionary

Default: None

Keyword arguments passed to the pipeline. The pipeline can override these arguments. Dependent on the backend.
Optional.

device

ComponentDevice

Default: None

The device on which the model is loaded. If None, the default device is automatically selected. If a device or a device map is specified in pipeline_kwargs, it overrides this parameter (only applicable to the Hugging Face backend).
Optional.