MCPcopy Create free account

hub / github.com/billwuhao/ComfyUI_ACE-Step / functions

Functions348 in github.com/billwuhao/ComfyUI_ACE-Step

↓ 12 callersMethodencode
(self, x)
ace_step/music_dcae/music_vocoder.py:556
↓ 11 callersMethod_addwords
(self, words,language,text,score,symbol=None)
ace_step/language_segmentation/LangSegment.py:370
↓ 9 callersFunctionnum2str
(value_string: str)
ace_step/language_segmentation/utils/num.py:299
↓ 8 callersFunctionget_padding
(kernel_size, dilation=1)
ace_step/music_dcae/music_vocoder.py:266
↓ 8 callersMethodload
(self, models, lora_name, lora_weight)
ace_step_nodes.py:340
↓ 7 callersMethod__init__
Construct an linear object.
ace_step/ace_models/lyrics_utils/lyric_encoder.py:822
↓ 7 callersMethod__init__
(self, drop_prob: float = 0.0, scale_by_keep: bool = True)
ace_step/music_dcae/music_vocoder.py:53
↓ 7 callersMethoddecode
(self, mel)
ace_step/music_dcae/music_vocoder.py:550
↓ 7 callersFunctionnum2chn
( number_string, numbering_type=NUMBERING_TYPES[1], big=False, traditional=False, alt_zero
ace_step/ace_models/lyrics_utils/zh_num2words.py:649
↓ 5 callersMethodcardinal2chntext
(self)
ace_step/ace_models/lyrics_utils/zh_num2words.py:764
↓ 5 callersMethodgetTexts
(self, text:str)
ace_step/language_segmentation/LangSegment.py:745
↓ 4 callersMethod_get_filters_string
(self)
ace_step/language_segmentation/LangSegment.py:428
↓ 4 callersFunctioncache_audio_tensor
( cache_dir, audio_tensor: torch.Tensor, sample_rate: int, filename_prefix: str = "cached_audi
ace_step_nodes.py:33
↓ 4 callersFunctionchn2num
(chinese_string, numbering_type=NUMBERING_TYPES[1])
ace_step/ace_models/lyrics_utils/zh_num2words.py:567
↓ 4 callersMethodgetCounts
(self)
ace_step/language_segmentation/LangSegment.py:733
↓ 4 callersFunctionset_all_seeds
(seed)
ace_step_nodes.py:56
↓ 4 callersMethodsetfilters
(self, filters)
ace_step/language_segmentation/LangSegment.py:715
↓ 3 callersMethod__init__
( self, height=16, width=4096, patch_size=(16, 1), in_channels=8,
ace_step/ace_models/ace_step_transformer.py:156
↓ 3 callersMethod__init__
(self, date=None, chntext=None)
ace_step/ace_models/lyrics_utils/zh_num2words.py:836
↓ 3 callersMethod_cleans_text
(self, cleans_text)
ace_step/language_segmentation/LangSegment.py:404
↓ 3 callersFunction_expand_currency
(m, lang="en", currency="USD")
ace_step/ace_models/lyrics_utils/lyric_tokenizer.py:487
↓ 3 callersMethod_lang_classify
(self, cleans_text)
ace_step/language_segmentation/LangSegment.py:420
↓ 3 callersFunctionapg_forward
( pred_cond: torch.Tensor, # [B, C, H, W] pred_uncond: torch.Tensor, # [B, C, H, W] guidance_sca
ace_step/apg_guidance.py:30
↓ 3 callersMethodapply_rotary_emb
Apply rotary embeddings to input tensors using the given frequency tensor. This function applies rotary embeddings to the given query
ace_step/ace_models/customer_attention_processor.py:34
↓ 3 callersMethodapply_rotary_emb
Apply rotary embeddings to input tensors using the given frequency tensor. This function applies rotary embeddings to the given query
ace_step/ace_models/customer_attention_processor.py:213
↓ 3 callersMethodbatch_decode
Convert a list of lists of token ids into a list of strings by calling decode. Args: sequences (`Union[List[int], List[L
ace_step/ace_models/lyrics_utils/lyric_tokenizer.py:677
↓ 3 callersFunctioncfg_forward
(cond_output, uncond_output, cfg_strength)
ace_step/apg_guidance.py:56
↓ 3 callersMethodcreate
(cls, index, value, numbering_type=NUMBERING_TYPES[1], small_unit=False)
ace_step/ace_models/lyrics_utils/zh_num2words.py:431
↓ 3 callersMethoddecode
(self, latents, audio_lengths=None, sr=None)
ace_step/music_dcae/music_dcae_pipeline.py:115
↓ 3 callersMethodget_text_embeddings
(self, texts, device, text_max_length=256)
ace_step/pipeline_ace_step.py:89
↓ 3 callersFunctionlowercase
(text)
ace_step/ace_models/lyrics_utils/lyric_tokenizer.py:545
↓ 3 callersFunctionval2tuple
Return tuple with min_len by repeating element at idx_repeat.
ace_step/ace_models/attention.py:42
↓ 2 callersMethod__init__
( self, in_dim: int, out_dim: int, kernel_size=3, stride=1, di
ace_step/ace_models/attention.py:66
↓ 2 callersMethod_clear_text_number
(self, text)
ace_step/language_segmentation/LangSegment.py:334
↓ 2 callersMethod_clears
(self)
ace_step/language_segmentation/LangSegment.py:294
↓ 2 callersMethod_init_step_index
(self, timestep)
ace_step/schedulers/scheduling_flow_match_euler_discrete.py:236
↓ 2 callersMethod_init_step_index
(self, timestep)
ace_step/schedulers/scheduling_flow_match_heun_discrete.py:197
↓ 2 callersMethod_parse_language
(self, words , segment)
ace_step/language_segmentation/LangSegment.py:432
↓ 2 callersMethod_process_tags
(self, words , text , root_tag)
ace_step/language_segmentation/LangSegment.py:625
↓ 2 callersMethod_saveData
(self, words,language:str,text:str,score:float,symbol=None)
ace_step/language_segmentation/LangSegment.py:340
↓ 2 callersMethod_set_cos_sin_cache
(self, seq_len, device, dtype)
ace_step/ace_models/ace_step_transformer.py:70
↓ 2 callersMethod_sigma_to_t
(self, sigma)
ace_step/schedulers/scheduling_flow_match_euler_discrete.py:170
↓ 2 callersMethod_sigma_to_t
(self, sigma)
ace_step/schedulers/scheduling_flow_match_pingpong.py:170
↓ 2 callersMethod_sigma_to_t
(self, sigma)
ace_step/schedulers/scheduling_flow_match_heun_discrete.py:140
↓ 2 callersMethod_statistics
(self, language, text)
ace_step/language_segmentation/LangSegment.py:324
↓ 2 callersMethodcalc_v
( self, zt_src, zt_tar, t, encoder_text_hidden_states, text_at
ace_step/pipeline_ace_step.py:230
↓ 2 callersMethodclassify
(self, text:str)
ace_step/language_segmentation/LangSegment.py:760
↓ 2 callersFunctioncollapse_whitespace
(text)
ace_step/ace_models/lyrics_utils/lyric_tokenizer.py:549
↓ 2 callersMethodcompress
(self, x: Tensor)
ace_step/music_dcae/music_log_mel.py:93
↓ 2 callersFunctioncreate_system
根据数字系统类型返回创建相应的数字系统,默认为 mid NUMBERING_TYPES = ['low', 'mid', 'high']: 中文数字系统类型 low: '兆' = '亿' * '十' = $10^{9}$, '京' = '兆' * '十', et
ace_step/ace_models/lyrics_utils/zh_num2words.py:531
↓ 2 callersMethoddigit2chntext
(self)
ace_step/ace_models/lyrics_utils/zh_num2words.py:780
↓ 2 callersFunctionexpand_abbreviations_multilingual
(text, lang="en")
ace_step/ace_models/lyrics_utils/lyric_tokenizer.py:237
↓ 2 callersFunctionexpand_numbers_multilingual
(text, lang="en")
ace_step/ace_models/lyrics_utils/lyric_tokenizer.py:524
↓ 2 callersFunctionexpand_symbols_multilingual
(text, lang="en")
ace_step/ace_models/lyrics_utils/lyric_tokenizer.py:433
↓ 2 callersMethodextend_pe
Reset the positional encodings.
ace_step/ace_models/lyrics_utils/lyric_encoder.py:738
↓ 2 callersMethodforward_attention
Compute attention context vector. Args: value (torch.Tensor): Transformed value, size (#batch, n_head, time2, d_k
ace_step/ace_models/lyrics_utils/lyric_encoder.py:232
↓ 2 callersMethodforward_qkv
Transform query, key and value. Args: query (torch.Tensor): Query tensor (#batch, time1, size). key (torch.Tensor): K
ace_step/ace_models/lyrics_utils/lyric_encoder.py:204
↓ 2 callersFunctionget_symbol
(char, system)
ace_step/ace_models/lyrics_utils/zh_num2words.py:568
↓ 2 callersMethodindex_for_timestep
(self, timestep, schedule_timesteps=None)
ace_step/schedulers/scheduling_flow_match_euler_discrete.py:222
↓ 2 callersMethodindex_for_timestep
(self, timestep, schedule_timesteps=None)
ace_step/schedulers/scheduling_flow_match_pingpong.py:222
↓ 2 callersMethodinfer_latents
(self, input_audio_path)
ace_step/pipeline_ace_step.py:1145
↓ 2 callersMethodload_json
(self, file_path)
ace_step/data_sampler.py:16
↓ 2 callersMethodpreprocess_text
(self, txt, lang)
ace_step/ace_models/lyrics_utils/lyric_tokenizer.py:641
↓ 2 callersFunctionsample_data
(json_data)
ace_step_nodes.py:75
↓ 2 callersMethodset_seeds
(self, batch_size, manual_seeds=None)
ace_step/pipeline_ace_step.py:148
↓ 2 callersFunctionsubsequent_chunk_mask
Create mask for subsequent steps (size, size) with chunk size, this is for streaming encoder Args: size (int): size of mask
ace_step/ace_models/lyrics_utils/lyric_encoder.py:473
↓ 2 callersMethodtelephone2chntext
(self, fixed=False)
ace_step/ace_models/lyrics_utils/zh_num2words.py:801
↓ 2 callersMethodto_chinese_number
(self, num:str)
ace_step/language_segmentation/LangSegment.py:175
↓ 2 callersMethodtokenize_lyrics
(self, lyrics, debug=False)
ace_step/pipeline_ace_step.py:196
↓ 2 callersFunctionverbalize_digit
(value_string: str, alt_one=False)
ace_step/language_segmentation/utils/num.py:291
↓ 1 callersMethod__init__
( self, n_fft=2048, win_length=2048, hop_length=512, center=False,
ace_step/music_dcae/music_log_mel.py:8
↓ 1 callersMethod__str__
(self)
ace_step/ace_models/lyrics_utils/zh_num2words.py:407
↓ 1 callersFunction_expand_decimal_point
(m, lang="en")
ace_step/ace_models/lyrics_utils/lyric_tokenizer.py:482
↓ 1 callersFunction_expand_number
(m, lang="en")
ace_step/ace_models/lyrics_utils/lyric_tokenizer.py:520
↓ 1 callersFunction_expand_ordinal
(m, lang="en")
ace_step/ace_models/lyrics_utils/lyric_tokenizer.py:516
↓ 1 callersMethod_format_chinese_data
(self, date_str:str)
ace_step/language_segmentation/LangSegment.py:124
↓ 1 callersMethod_get_prev_data
(self, words)
ace_step/language_segmentation/LangSegment.py:392
↓ 1 callersFunction_get_value
(value_string: str, use_zero: bool=True)
ace_step/language_segmentation/utils/num.py:256
↓ 1 callersMethod_init_step_index
(self, timestep)
ace_step/schedulers/scheduling_flow_match_pingpong.py:236
↓ 1 callersMethod_insert_english_uppercase
(self, word)
ace_step/language_segmentation/LangSegment.py:316
↓ 1 callersMethod_is_chinese
(self, word)
ace_step/language_segmentation/LangSegment.py:305
↓ 1 callersMethod_is_japanese_kana
(self, word)
ace_step/language_segmentation/LangSegment.py:311
↓ 1 callersMethod_match_ending
(self, input , index)
ace_step/language_segmentation/LangSegment.py:397
↓ 1 callersMethod_mean_processing
(self, text:str)
ace_step/language_segmentation/LangSegment.py:409
↓ 1 callersMethod_merge_results
(self, words)
ace_step/language_segmentation/LangSegment.py:638
↓ 1 callersMethod_parse_symbols
(self, text)
ace_step/language_segmentation/LangSegment.py:650
↓ 1 callersMethod_pattern_symbols
(self, item , text)
ace_step/language_segmentation/LangSegment.py:520
↓ 1 callersMethod_process_symbol_SSML
(self, words,data)
ace_step/language_segmentation/LangSegment.py:487
↓ 1 callersMethod_split_camel_case
(self, word)
ace_step/language_segmentation/LangSegment.py:321
↓ 1 callersMethodadd_latents_noise
( self, gt_latents, sigma_max, noise, scheduler_type, infer_st
ace_step/pipeline_ace_step.py:533
↓ 1 callersFunctionadd_optional_chunk_mask
Apply optional mask for encoder. Args: xs (torch.Tensor): padded input, (B, L, D), L for max length mask (torch.Tensor): mask for
ace_step/ace_models/lyrics_utils/lyric_encoder.py:511
↓ 1 callersFunctionbasic_cleaners
Basic pipeline that lowercases and collapses whitespace without transliteration.
ace_step/ace_models/lyrics_utils/lyric_tokenizer.py:576
↓ 1 callersFunctioncfg_double_condition_forward
( cond_output, uncond_output, only_text_cond_output, guidance_scale_text, guidance_scale_l
ace_step/apg_guidance.py:60
↓ 1 callersFunctioncfg_zero_star
(noise_pred_with_cond, noise_pred_uncond, guidance_scale, i, zero_steps=1, use_zero_init=True)
ace_step/apg_guidance.py:84
↓ 1 callersMethodcheck_input_length
(self, txt, lang)
ace_step/ace_models/lyrics_utils/lyric_tokenizer.py:633
↓ 1 callersFunctionchinese_transliterate
(text)
ace_step/ace_models/lyrics_utils/lyric_tokenizer.py:583
↓ 1 callersMethodcollate_fn
Collate function for DataLoader Args: batch: List of examples Returns: dict: Collated batch
ace_step/text2music_dataset.py:588
↓ 1 callersFunctioncompute_value
Compute the value. When current unit is larger than previous unit, current unit * all previous units will be used as all previous uni
ace_step/ace_models/lyrics_utils/zh_num2words.py:619
↓ 1 callersFunctioncorrect_symbols
一百八 to 一百八十 一亿一千三百万 to 一亿 一千万 三百万
ace_step/ace_models/lyrics_utils/zh_num2words.py:587
↓ 1 callersFunctioncross_norm
(hidden_states, controlnet_input)
ace_step/ace_models/ace_step_transformer.py:32
next →1–100 of 348, ranked by callers