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

모델 학습 중에 다음과 같은 오류가 발생했다. 

 

뭔가 해서 계속 구글링을 해봤는데, transformer 버전 변경에 따른 오류였다.. 

사실 필자의 잘못인 게, 돌린 모델의 requirements를 설정해주지 않고 그냥 내 맘대로 돌리려다 보니 발생한 문제였다.. transformer 3.xx 버전을 요구하는 모델인데, 본인은 그냥 새로 설치하고 4.xx 버전을 사용했더니 위와 같이 오류 발생.. requirements 파일 대로 다시 환경 설정을 해주어도 되지만(그게 맞지만) 우선 아래와 같이 인자를 추가한 후 실행하니 잘 돌아가서 그냥 쓰련다. ㅎㅎ

 

핵심은 4.xx버전에서는 pre-trained 모델 불러올 때 인자로 return-dict를 false로 설정해주는 것이다. 

 

https://huggingface.co/docs/transformers/migration

Migrating from previous packages