o 5heeJ@sTddlZddlZddlmZmZddlmZdZeeZe Z ej j fddZ dS)N)DebertaV2ModelDebertaV2Tokenizer)configzmicrosoft/deberta-v2-xlargec CsJtjdkrtjjr|dkrd}|sd}|tvr%t t  |t|<t 9t |dd}|D] }|| |||<q2t|di|ddi}t|d d d d d }Wdn1sewYt||jd ksJ||jd t|f|}g}tt|D]}||||d}||qtj|d d}|jS)Ndarwincpumpscudapt)return_tensorsoutput_hidden_statesT hidden_statesr)dim)sysplatformtorchbackendsr is_availablemodelskeysrfrom_pretrained LOCAL_PATHtono_grad tokenizercatrlenshaperangerepeatappendT) textword2phdeviceinputsires word2phonephone_level_featurerepeat_featurerr+D:\Code\tts-9nine\text\english_bert_mock.pyget_bert_features0     "* r0)rr transformersrrrrrrdictrbert_gen_configr(r0rrrr/s