o ´‹ jx ã@sddlZdd„ZdS)éNcCs.tj|t d¡dd}i}| ¡D]\}}d|vr$||| dd¡<qd|vr1||| dd¡<qd|vr>||| dd ¡<qd |vrK||| d d ¡<qd |vrX||| d d ¡<qd |vre||| d d¡<qd|vrr||| dd¡<qd|vr||| dd¡<qd|vrŒ||| dd¡<qd|vr™||| dd¡<qd|vr¦||| dd¡<qd|vr³||| dd¡<qd|vrÀ||| dd¡<qd|vrÍ||| dd¡<qd|vrÚ||| dd ¡<qd!|vrç||| d!d"¡<qd#|vrô||| d#d$¡<qd%|vr||| d%d&¡<qd'|vr||| d'd(¡<q|||<q|S))NÚcpu)Ú map_locationÚmodelzenc_p.z text_encoder.zdec.zwaveform_decoder.zenc_q.zposterior_encoder.z flow.flows.2.z flow.flows.1.z flow.flows.4.z flow.flows.6.z flow.flows.3.z dp.flows.0.mz&duration_predictor.flows.0.translationzdp.flows.0.logsz$duration_predictor.flows.0.log_scalez dp.flows.1zduration_predictor.flows.1z dp.flows.3zduration_predictor.flows.2z dp.flows.5zduration_predictor.flows.3z dp.flows.7zduration_predictor.flows.4zdp.post_flows.0.mz+duration_predictor.post_flows.0.translationzdp.post_flows.0.logsz)duration_predictor.post_flows.0.log_scalezdp.post_flows.1zduration_predictor.post_flows.1zdp.post_flows.3zduration_predictor.post_flows.2zdp.post_flows.5zduration_predictor.post_flows.3zdp.post_flows.7zduration_predictor.post_flows.4zdp.zduration_predictor.)ÚtorchÚloadÚdeviceÚitemsÚreplace)Úcheckpoint_fileÚchkÚnew_chkÚkÚv©rúH/home/kuhnn/.local/lib/python3.10/site-packages/TTS/tts/utils/fairseq.pyÚrehash_fairseq_vits_checkpointsV   r)rrrrrrÚs