ChR vSSKJr SSKrSSKrSSKJrJr SSKJr \R"\ 5r "SS\5r g)) annotationsN)Tensornn)Modulecd^\rSrSr%Sr/SQrS\S'S S U4SjjjrSSjrSS .SS jjr S r U=r $) WordWeights zDThis model can weight word embeddings, for example, with idf-values.)vocab word_weightsunknown_word_weight list[str] config_keysc>[TU]5 XlX lX0l/nSnUHLnUnXb;aX&nO,UR 5U;aX&R 5nOUS- nUR U5 MN [RUS[U5SU35 [R"[U5S5Ul URRS[R"U5R!S505 g)a Initializes the WordWeights class. Args: vocab (List[str]): Vocabulary of the tokenizer. word_weights (Dict[str, float]): Mapping of tokens to a float weight value. Word embeddings are multiplied by this float value. Tokens in word_weights must not be equal to the vocab (can contain more or less values). unknown_word_weight (float, optional): Weight for words in vocab that do not appear in the word_weights lookup. These can be, for example, rare words in the vocab where no weight exists. Defaults to 1. rz of z0 words without a weighting value. Set weight to weightN)super__init__r r r lowerappendloggerinfolenr Embedding emb_layerload_state_dicttorch FloatTensor unsqueeze) selfr r r weightsnum_unknown_wordswordr __class__s `/var/www/html/shao/venv/lib/python3.13/site-packages/sentence_transformers/models/WordWeights.pyrWordWeights.__init__s  (#6 D(F#%+-%jjl3!Q&! NN6 "   !c%j\1abuav w c%j!4 &&%2C2CG2L2V2VWX2Y'Z[c<USnUSnURUS5RS5nXBR5-n[R"US5nUR S5R UR55nX7-nURX6S.5 U$)Nattention_masktoken_embeddings input_idsr)r)token_weights_sum) rsqueezefloatrsumrexpandsizeupdate)rfeaturesr(r)token_weights_raw token_weightsr,token_weights_expandeds r$forwardWordWeights.forward5s!"23#$67!NN8K+@AII"M),@,@,BB !IImQ7"/!8!8!rRs0" 6   8 $9&&9&r&