Hub
NodeNode / Source

BERT Model Selector

The node allows downloading the model available on TensorFlow Hub and HuggingFace. The trusted models are added to the lists. For HuggingFace it is possible to paste the model name into the selector. In case you would like to test other models please refer to Advanced tab. The model can be cached on a disk.

Node details

Output ports
  1. Type: se.redfield.bert.nodes.port.BertModelPortObject
    BERT Model
    The BERT model that was downloaded or read from cache.

Extension

The BERT Model Selector node is part of this extension: