U 1szCNN.forward..)Z transposertorchcatupdate)rfeaturesZvectorsoutrr$r forward-s  z CNN.forward)returncCs|jSN)rrrrr get_word_embedding_dimension7sz CNN.get_word_embedding_dimension)textr,cKs tdSr-)NotImplementedError)rr0kwargsrrr tokenize:sz CNN.tokenizeT) output_pathsafe_serializationc Cslttj|dd}tj||ddW5QRX|rNt|tj|dnt | tj|ddS)Ncnn_config.jsonwr)indentmodel.safetensorspytorch_model.bin) openospathjoinjsondumpget_config_dictsave_safetensors_modelr&saveZ state_dict)rr4r5ZfOutrrr rC=s zCNN.savecsfddjDS)Ncsi|]}|j|qSr)__dict__)r#keyr.rr Gsz'CNN.get_config_dict..)rr.rr.r rAFszCNN.get_config_dict) input_pathc Csttj|dd}t|}W5QRXtf|}tjtj|dr`t|tj|dn$| t jtj|dt dd|S)Nr6rr9r:cpu)Z map_location) r;r<r=r>r?loadrexistsload_safetensors_modelZload_state_dictr&Zdevice)rGZfInconfigmodelrrr rJIs zCNN.load)T)__name__ __module__ __qualname____doc__rrrr+r/strr3boolrCrA staticmethodrJrrrr r s"   r) r?r<typingrr&Zsafetensors.torchrrLrrBrrrrrrr s