Error

[Astudnew]

Hi
I clone your repo on google colab to run it
At the beginning I get an error that torchcrf module not found but after I install pytorch-crf (!pip install pytorch-crf==0.4.0) it works , but I get other errors :
"Traceback (most recent call last):
File "bert_crf.py", line 433, in
t_loss, v_loss = train(train_iter, eval_iter, tag2idx, config=config, bert_model=config.bert_model)
File "bert_crf.py", line 196, in train
model = Bert_CRF.from_pretrained(bert_model, num_labels = len(tag2idx))
File "/usr/local/lib/python3.6/dist-packages/transformers/modeling_utils.py", line 1034, in from_pretrained
model = cls(config, *model_args, **model_kwargs)
File "bert_crf.py", line 141, in init
self.crf = CRF(self.num_labels, batch_first=True)
TypeError: init() got an unexpected keyword argument 'batch_firs"
how I can fix them?