U =s z LSTM.savecsfddjDS)Ncsi|]}|j|qSr)__dict__).0keyr%rr Isz(LSTM.get_config_dict..)rr%rr%rr8HszLSTM.get_config_dict) input_pathc Csttj|dd}t|}W5QRXtf|}tjtj|dr`t|tj|dn$| t jtj|dt dd|S)Nr-rr0r1r)Z map_location) r2r3r4r5r6loadrexistsload_safetensors_modelZload_state_dictrr;)rCZfInconfigmodelrrrrEKs z LSTM.loadN)rrT)T)__name__ __module__ __qualname____doc__intfloatboolrr"r&strrr*r>r8 staticmethodrErrrrr s$   r) r6r3typingrrZsafetensors.torchrrGrr<rrrrrrrs