o i* @sddlZdejd<ddlmZddlZddlmmZddl Z ddl m Z ddl m Z ddl mZddlmZmZmZdd lmZdd lmZmZejrQd nejrXd n ejjr`d ndZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&e'e(ed)dedZ*e de*j+j,Z-e*j+j.Z/e*j0j1Z2e*j+j3Z3e*j+j4j5Z5e*j+j4j6Z6e*j+j4j7Z7e*j+j4j8Z8e*j+j4j9Z9e*j+j4j:Z:e(e dede dZ;dZd!Z?d"d#gd$d%ggZ@d&d'gZAd(ZBejCDe<seEe<d)ZFe5d*krd+ZGne5d,krd-ZGee5eFeGd.ZHee2e3\ZIZJee-dCie/eJe7d/eKe:e8e9e7e6e5d0eKe#d1eId2LeZ+e5d,kr1ejMndZNee+e;eeNeBd3Z+e 'e=\ZOZPeOjQdd'krSejReOdd(d4ZOeSeReTeOZUeUe&krieOe&eUZOePe6krye jVWePe6ZXeXeOZOeOLeZOeYe+4eOZZeZ[dd5d'ZZWdn 1swYdZ\ej]d'de7ed6Z^ej]d'dej_ed7Z`eAdureAandZbe@D]iZcec\ZdZeebdureeednebfdZgehede6e8Zieheee6e8Zjehege6e8Zkeie\Zlejme^eZdde\eiddfej]d'eke7ed6fd'd8Z^ejme`ejnd'elej_ed7ej]d'ekej_ed7fd9d8Z`ejZ\qejme^eZdde\dddffd'd8Z^ejoe`de^jQd'e`jQd9fd(d:Z`e?gZpe3d;kr^eepZqnepgZqerd\ZuZverd?eujQeuLejMZueu[dd5d'Zwe5d*kreHxewyZzne5d,kreHew{dyZzeUe&krezeUe&Zzeewdy|e<d@e }e<dAeze6erdBezjQWddS1swYdS)DN1PYTORCH_ENABLE_MPS_FALLBACK)files) cached_path) get_class) OmegaConf)load_checkpoint load_vocodersave_spectrogram)CFM)convert_char_to_pinyin get_tokenizercudaxpumpscpu F5TTS_v1_Basei g@eulergg?g?f5_ttszconfigs/z.yamlz f5_tts.model.zhf://SWivid/F5-TTS/z/model_z .safetensorstestsz%infer/examples/basic/basic_ref_en.wavz2Some call me nature, others call me mother nature.z.Some call me optimist, others call me realist.gQ?gQ@g)\(@g@g333333?TFvocosz'../checkpoints/charactr/vocos-mel-24khzbigvganz,../checkpoints/bigvgan_v2_24khz_100band_256x) vocoder_nameis_local local_path)text_num_embedsmel_dim)n_fft hop_length win_lengthn_mel_channelstarget_sample_rate mel_spec_type)method) transformermel_spec_kwargs odeint_kwargsvocab_char_map)dtypeuse_ema)dimkeepdim)device)r*r/)r,)valuepinyinztext : zpinyin: )condtextdurationsteps cfg_strengthsway_sampling_coefseed edit_maskzGenerated mel: z/speech_edit_out.pngz/speech_edit_out.wavzGenerated wav: )~osenvironimportlib.resourcesrtorchtorch.nn.functionalnn functionalF torchaudior hydra.utilsr omegaconfrf5_tts.infer.utils_inferrr r f5_tts.modelr f5_tts.model.utilsr r r is_availablerbackendsrr/r9exp_name ckpt_stepnfe_stepr7 ode_methodr8speed target_rmsloadstrjoinpath model_cfgmodelbackbone model_clsarch model_arcdatasetsname dataset_name tokenizermel_specr$r#r"r r!r ckpt_path output_dir audio_to_edit origin_text target_text parts_to_edit fix_durationr+pathexistsmakedirslocalvocoder_local_pathvocoderr) vocab_sizedicttofloat32r*audiosrshapemeansqrtsquarerms transformsResample resamplerinference_mode original_melpermute offset_framezerosmel_condboolr:copy fix_dur_listpartstartendpop part_dur_secround start_frame end_framepart_dur_frames keep_framescatonespad text_listfinal_text_listprintr5sample generated trajectory gen_mel_specdecodergenerated_wavesqueezenumpysaver;r;r;L/home/ubuntu/.local/lib/python3.10/site-packages/f5_tts/infer/speech_edit.pys2                        ($          $