[KoBERT 에러] TypeError: dropout(): argument 'input' (position 1) must be Tensor, not str 해결 방법

KoBERT 사용 도중 만난 에러

분명 model에 input되는 parameters Type이 모두 Tensor였음에도 불구하고 Tensor로 입력하라는 에러....

 

해결 방법

의외로 간단하다.

transformers 설치 시,

기존 pip install transformers로 했다면

pip install transformers==3 으로 다시 설치해보자

주 의

Colab에서 사용한다면,

!pip install transformers==3 이후에, 런타임 초기화 후 재실행 해야한다.