jinmang2/DOOLY

transformers model weights load error

jinmang2 opened this issue · 3 comments

huggingface/transformers@5da33f8

4월 12일에 해당 스크립트보고 4.18.0 대비 커밋을 푸쉬했는데 (89ce3f9)

4월 15일에 메서드에 arguments가 하나 더 추가되었다... ㅎㅎ (loaded_state_dict_keys)

v1.x.x에서는 modeling_utils를 건드리지 않고 model weights를 호출해야할 방법을 찾아야겠다.

우선 이를 v0.1.3에서 fix한다.

  • pypi에서 remove했는데 자꾸 에러가 발생해서 0.1.2는 pass...
  • 어차피 v1.x 리뉴얼을 준비 중이라 상관없지만
  • 이후에 또다른 issue를 확인해야겠다

다행히 v4.19.0 version으로 디버깅이 가능할 듯 하다.

version.parse(transformers.__version__) >= version.parse("4.19.0")

하지만... transformers의 backend 모듈에 디펜던시가 생기면 5버전 5.x버전 업데이트하면서 해당 모듈에 계속 버그가 생길 것 같다. PR 때리던지 해야지 원 참... 왜 model에는 subfolder 지원을 안해주는걸까!

!pip install dooly
# !pip install git+https://github.com/jinmang2/DOOLY

import dooly
print(dooly.__version__)
# v0.1.3

from dooly import Dooly

mt = Dooly("mt", "multi")

mt("L, 알고 있는가? 사신은 사과만 먹는다네.", "ko", "en")
# L, do you know? You only eat apples.
  • 이 와중에 번역 이상한거 실화...?

bug fix 및 v0.1.3 배포. issue closing