No SmartTokenPooling implementation file (ModuleNotFoundError: No module named 'sbert_patch') ?

#1
by zegri - opened

Hi! I'm trying to test out this model, but get
'ModuleNotFoundError: No module named 'sbert_patch''
when trying to init the model model = SentenceTransformer("jensjorisdecorte/ConTeXT-Skill-Extraction-base"). From what I understand by looking at the model repo and the exception, there should be a 'sbert_patch.SmartTokenPooling.SmartTokenPooling' layer implementation file?

Using:
sentence-transformers = "3.1.0"
transformers = "4.44.2"
Also tried with:
sentence-transformers = "^3.3.1"
transformers = "^4.48.0"

If it's not the case of intending to not release this yet, could you please help me out? I want to try out this model for work on a potential research paper ( your work has been of huge help already :) ).

Thanks a lot!

Sign up or log in to comment