It goes under ConveRTFeaturizer for anyone needing this.
EX:
pipeline:
- name: ConveRTTokenizer
- name: ConveRTFeaturizer
model_url: "https://github.com/connorbrinton/polyai-models/releases/download/v1.0/model.tar.gz"
- name: DIETClassifier
epochs: 200