불러올 때마다 No sentence-transformers model found with name nlpai-lab/KoE5. Creating a new one with mean pooling. 라고 나옵니다

#2
by Gyeongmo - opened

안녕하세요?

이 모델을 불러올 때마다 No sentence-transformers model found with name nlpai-lab/KoE5. Creating a new one with mean pooling. 라는 메시지가 나옵니다. 이 경고 메시지가 나와도 작동은 잘 되는데요, 왜 그런지 이유를 모르겠습니다.

Sentence-BERT 버전은 3.2.0입니다.

이 메시지를 나오지 않게 하려면 어떻게 해야 하나요?

NLP & AI - Korea University org
edited Oct 14

안녕하세요, sentence-transformers의 경우 huggingface model 위에 추가적인 레이어(mean pooling 등)를 포함하여 구성합니다.
현재 KoE5 리포지터리에 추가적인 레이어에 대한 설정 파일(config_sentence_transformers.json)이 누락되어 있어 경고 메시지가 나오고 있습니다.
설정 파일을 추가하여 업데이트 하도록 하겠습니다.

(참고로, KoE5의 경우 mean pooling으로 학습하였으므로 경고메시지가 나오더라도 inference에 문제가 있는 것은 아닙니다.)

NLP & AI - Korea University org

https://huggingface.co./nlpai-lab/KoE5/commit/17ffe044a3fcc212b8a44c1bc1a476b95dd31f1e

sentence transformers config 추가하였습니다. 다시 로드하시면 warning 없이 진행 될 것입니다.

taeminlee changed discussion status to closed

https://huggingface.co./nlpai-lab/KoE5/commit/17ffe044a3fcc212b8a44c1bc1a476b95dd31f1e

sentence transformers config 추가하였습니다. 다시 로드하시면 warning 없이 진행 될 것입니다.

이제는 경고 메시지 없이 잘 작동합니다. 좋은 한국어 Sentence-BERT 모델을 공유해 주셔서 감사합니다.

Sign up or log in to comment