BERT Classification Learner error: 'str' object has no attribute 'shape'

Hi @Redfield,

I have tried several models from Hugging Face and I got this error:

ERROR BERT Classification Learner 3:2        Execute failed: 'str' object has no attribute 'shape'
Traceback (most recent call last):
  File "<string>", line 15, in <module>
  File "C:\Program Files\KNIME\plugins\se.redfield.bert_0.2.0.v202111180712\py\BertClassifier.py", line 93, in run_train
    embedder = BertEmbedder.from_pretrained(model_type, bert_model_handle, sentence_column, max_seq_length=max_seq_length, cache_dir=cache_dir)
  File "C:\Program Files\KNIME\plugins\se.redfield.bert_0.2.0.v202111180712\py\BertEmbedder.py", line 95, in from_pretrained
    return BertEmbedder(bert_layer, tokenizer)
  File "C:\Program Files\KNIME\plugins\se.redfield.bert_0.2.0.v202111180712\py\BertEmbedder.py", line 20, in __init__
    if(pooled_output.shape.ndims == 3 and sequence_output.shape.ndims == 2):
AttributeError: 'str' object has no attribute 'shape'

The models I have tried:
xlm-roberta-base
bert-base-multilingual-cased
HooshvareLab/bert-fa-zwnj-base

How can I fix this?