반응형
에러 발생: Config Assertion Error: config has to be initialized..
Config assertion error in transformer
File "/root/miniconda3/envs/lib/python3.10/site-packages/transformers/modeling_utils.py", line 3916, in save_pretrained misplaced_generation_parameters = model_to_save.config._get_non_default_generation_parameters()
assert "encoder" in kwargs, "Config has to be initialized with encoder config" AssertionError: Config has to be initialized with encoder config
trainer.model.save_pretrained(self.output_dir)
trainer.model.save_pretrained(self.output_dir)
save_pretrained함수를 실행했었다. 하지만
Transformer의 trainer를 사용하는 도중 assertion error가 발생했다
패키지 문제일 것으로 생각해서 여러 transformer 버전을 시도해보았다
pip install transformers==4.40.1
.
pip install transformers==4.40.1
pip install transformers==4.40.1
위의 버전을 설치했을 때 Assertion Error가 사라졌다
Reference
https://github.com/IBM/fastfit/issues/25
반응형