MCPcopy Create free account

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

Functions348 in github.com/billwuhao/ComfyUI_ACE-Step

↓ 1 callersMethoddate2chntext
(self)
ace_step/ace_models/lyrics_utils/zh_num2words.py:864
↓ 1 callersMethoddecode
( self, hidden_states: torch.Tensor, attention_mask: torch.Tensor, encoder_hid
ace_step/ace_models/ace_step_transformer.py:414
↓ 1 callersMethoddecode
(self, seq, skip_special_tokens=False)
ace_step/ace_models/lyrics_utils/lyric_tokenizer.py:666
↓ 1 callersMethoddecode_overlap
Decodes latents into waveforms using an overlapped DCAE and Vocoder.
ace_step/music_dcae/music_dcae_pipeline.py:147
↓ 1 callersFunctiondrop_path
Drop paths (Stochastic Depth) per sample (when applied in main path of residual blocks). This is the same as the DropConnect impl I created for E
ace_step/music_dcae/music_vocoder.py:25
↓ 1 callersMethodencode
( self, encoder_text_hidden_states: Optional[torch.Tensor] = None, text_attention_mask
ace_step/ace_models/ace_step_transformer.py:376
↓ 1 callersMethodencode
(self, audios, audio_lengths=None, sr=None)
ace_step/music_dcae/music_dcae_pipeline.py:77
↓ 1 callersMethodflowedit_diffusion_process
( self, encoder_text_hidden_states, text_attention_mask, speaker_embds,
ace_step/pipeline_ace_step.py:324
↓ 1 callersMethodforward_layers
( self, xs: torch.Tensor, chunk_masks: torch.Tensor, pos_emb: torch.Tensor,
ace_step/ace_models/lyrics_utils/lyric_encoder.py:1013
↓ 1 callersMethodforward_layers_checkpointed
( self, xs: torch.Tensor, chunk_masks: torch.Tensor, pos_emb: torch.Tensor,
ace_step/ace_models/lyrics_utils/lyric_encoder.py:1025
↓ 1 callersMethodforward_lyric_encoder
( self, lyric_token_idx: Optional[torch.LongTensor] = None, lyric_mask: Optional[torch
ace_step/ace_models/ace_step_transformer.py:363
↓ 1 callersMethodforward_mel
(self, audios)
ace_step/music_dcae/music_dcae_pipeline.py:68
↓ 1 callersMethodfraction2chntext
(self)
ace_step/ace_models/lyrics_utils/zh_num2words.py:826
↓ 1 callersMethodget_audio
Load and preprocess audio file Args: item: Dataset item containing filename Returns: torch.T
ace_step/text2music_dataset.py:388
↓ 1 callersMethodget_full_features
Get full features for a dataset index Args: idx: Dataset index Returns: dict: Dictionary of
ace_step/text2music_dataset.py:522
↓ 1 callersFunctionget_lang
(text, default_lang, threshold)
text2lyric.py:31
↓ 1 callersMethodget_lang
Detect the language of a text Args: text: Input text Returns: tuple: (primary_language, lang
ace_step/text2music_dataset.py:231
↓ 1 callersMethodget_lang
(self, text)
ace_step/pipeline_ace_step.py:184
↓ 1 callersFunctionget_same_padding
(kernel_size: Union[int, Tuple[int, ...]])
ace_step/ace_models/attention.py:58
↓ 1 callersFunctionget_spacy_lang
(lang)
ace_step/ace_models/lyrics_utils/lyric_tokenizer.py:23
↓ 1 callersMethodget_speaker_emb_file
Load speaker embedding file Args: speaker_emb_path: Path to speaker embedding file Returns:
ace_step/text2music_dataset.py:371
↓ 1 callersMethodget_text_embeddings_null
( self, texts, device, text_max_length=256, tau=0.01, l_min=8, l_max=10 )
ace_step/pipeline_ace_step.py:107
↓ 1 callersFunctionget_value
(value_string, use_zeros=True)
ace_step/ace_models/lyrics_utils/zh_num2words.py:660
↓ 1 callersMethodindex_for_timestep
(self, timestep, schedule_timesteps=None)
ace_step/schedulers/scheduling_flow_match_heun_discrete.py:183
↓ 1 callersFunctionis_silent_audio
Determine if an audio is silent and should be discarded Args: audio_tensor: torch.Tensor from torchaudio, shape (num_channels, n
ace_step/text2music_dataset.py:21
↓ 1 callersFunctionjapanese_cleaners
(text, katsu)
ace_step/ace_models/lyrics_utils/lyric_tokenizer.py:589
↓ 1 callersFunctionkorean_transliterate
(text)
ace_step/ace_models/lyrics_utils/lyric_tokenizer.py:595
↓ 1 callersMethodlatents2audio
(self, latents, target_wav_duration_second=30.0, sample_rate=48000)
ace_step/pipeline_ace_step.py:1129
↓ 1 callersMethodload_audio
(self, audio_path)
ace_step/music_dcae/music_dcae_pipeline.py:62
↓ 1 callersFunctionmain
()
ace_step/language_segmentation/LangSegment.py:780
↓ 1 callersMethodmoney2chntext
(self)
ace_step/ace_models/lyrics_utils/zh_num2words.py:901
↓ 1 callersFunctionmultilingual_cleaners
(text, lang)
ace_step/ace_models/lyrics_utils/lyric_tokenizer.py:553
↓ 1 callersFunctionnormalize_nsw
(raw_text)
ace_step/ace_models/lyrics_utils/zh_num2words.py:928
↓ 1 callersFunctionoptimized_scale
(positive_flat, negative_flat)
ace_step/apg_guidance.py:70
↓ 1 callersMethodpack_batch
Pack a batch of examples Args: batch: List of examples Returns: dict: Packed batch
ace_step/text2music_dataset.py:569
↓ 1 callersMethodpercentage2chntext
(self)
ace_step/ace_models/lyrics_utils/zh_num2words.py:924
↓ 1 callersMethodposition_encoding
For getting encoding in a streaming fashion Attention!!!!! we apply dropout only once at the whole utterance level in a none
ace_step/ace_models/lyrics_utils/lyric_encoder.py:787
↓ 1 callersFunctionprintList
功能:打印数组结果 기능: 어레이 결과 인쇄 機能:配列結果を印刷 Function: Print array results
ace_step/language_segmentation/LangSegment.py:763
↓ 1 callersMethodprocess
Process a dataset item into model-ready features Args: item: Dataset item Returns: list: Lis
ace_step/text2music_dataset.py:437
↓ 1 callersFunctionproject
( v0: torch.Tensor, # [B, C, H, W] v1: torch.Tensor, # [B, C, H, W] dims=[-1, -2], )
ace_step/apg_guidance.py:14
↓ 1 callersMethodrel_shift
Compute relative positional encoding. Args: x (torch.Tensor): Input tensor (batch, head, time1, 2*time1-1). time1 mea
ace_step/ace_models/lyrics_utils/lyric_encoder.py:348
↓ 1 callersFunctionremove_erhua
去除儿化音词中的儿: 他女儿在那边儿 -> 他女儿在那边
ace_step/ace_models/lyrics_utils/zh_num2words.py:1019
↓ 1 callersFunctionremove_space
(text)
ace_step/ace_models/lyrics_utils/zh_num2words.py:1046
↓ 1 callersMethodremove_weight_norm
(self)
ace_step/music_dcae/music_vocoder.py:355
↓ 1 callersMethodsample
(self)
ace_step/data_sampler.py:20
↓ 1 callersMethodsetup_full
Load and prepare the dataset Args: train: Whether this is a training dataset shuffle: Whether to shuffl
ace_step/text2music_dataset.py:207
↓ 1 callersMethodstep
Predict the sample from the previous timestep by reversing the SDE. This function propagates the diffusion process from the learned
ace_step/schedulers/scheduling_flow_match_pingpong.py:244
↓ 1 callersFunctionstring2symbols
(chinese_string, system)
ace_step/ace_models/lyrics_utils/zh_num2words.py:579
↓ 1 callersFunctiont2i_modulate
(x, shift, scale)
ace_step/ace_models/attention.py:54
↓ 1 callersFunctiontest_abbreviations_multilingual
()
ace_step/ace_models/lyrics_utils/lyric_tokenizer.py:810
↓ 1 callersFunctiontest_expand_numbers_multilingual
()
ace_step/ace_models/lyrics_utils/lyric_tokenizer.py:711
↓ 1 callersFunctiontest_symbols_multilingual
()
ace_step/ace_models/lyrics_utils/lyric_tokenizer.py:853
↓ 1 callersMethodtext2music_diffusion_process
( self, duration, encoder_text_hidden_states, text_attention_mask, spe
ace_step/pipeline_ace_step.py:576
↓ 1 callersMethodtime_shift
(self, mu: float, sigma: float, t: torch.Tensor)
ace_step/schedulers/scheduling_flow_match_euler_discrete.py:173
↓ 1 callersMethodtime_shift
(self, mu: float, sigma: float, t: torch.Tensor)
ace_step/schedulers/scheduling_flow_match_pingpong.py:173
↓ 1 callersMethodto_chinese_currency
(self, num:str)
ace_step/language_segmentation/LangSegment.py:196
↓ 1 callersMethodto_chinese_date
(self, num:str)
ace_step/language_segmentation/LangSegment.py:208
↓ 1 callersMethodto_chinese_telephone
(self, num:str)
ace_step/language_segmentation/LangSegment.py:189
↓ 1 callersFunctiontokenize_lyrics
(lyrics, default_lang, threshold)
text2lyric.py:47
↓ 1 callersMethodtokenize_lyrics
Tokenize lyrics into token indices Args: lyrics: Lyrics text debug: Whether to print debug information
ace_step/text2music_dataset.py:254
↓ 1 callersMethodtokenize_lyrics_map
Process and tokenize lyrics in a dataset item Args: item: Dataset item containing lyrics debug: Whether
ace_step/text2music_dataset.py:339
↓ 1 callersMethodunpatchfy
( self, hidden_states: torch.Tensor, width: int, )
ace_step/ace_models/ace_step_transformer.py:110
↓ 1 callersMethodupdate
(self, update_value: torch.Tensor)
ace_step/apg_guidance.py:9
↓ 1 callersFunctionval2list
Repeat `val` for `repeat_time` times and return the list or val if list/tuple.
ace_step/ace_models/attention.py:35
↓ 1 callersFunctionverbalize_cardinal
(value_string: str)
ace_step/language_segmentation/utils/num.py:274
MethodINPUT_TYPES
(s)
ace_step_nodes.py:111
MethodINPUT_TYPES
(cls)
ace_step_nodes.py:147
MethodINPUT_TYPES
(cls)
ace_step_nodes.py:168
MethodINPUT_TYPES
(cls)
ace_step_nodes.py:189
MethodINPUT_TYPES
(cls)
ace_step_nodes.py:324
MethodINPUT_TYPES
(cls)
ace_step_nodes.py:359
MethodINPUT_TYPES
(cls)
ace_step_nodes.py:432
MethodINPUT_TYPES
(cls)
ace_step_nodes.py:502
MethodINPUT_TYPES
(cls)
ace_step_nodes.py:569
MethodINPUT_TYPES
(cls)
text2lyric.py:108
Method__call__
( self, audio_duration: float = 60.0, prompt: str = None, negative_prompt: str
ace_step/pipeline_ace_step.py:1155
Method__call__
( self, attn: Attention, hidden_states: torch.FloatTensor, encoder_hidden_stat
ace_step/ace_models/customer_attention_processor.py:64
Method__call__
( self, attn: Attention, hidden_states: torch.FloatTensor, encoder_hidden_stat
ace_step/ace_models/customer_attention_processor.py:243
Method__call__
(self, text)
ace_step/ace_models/lyrics_utils/zh_num2words.py:1083
Method__enter__
(self)
ace_step/cpu_offload.py:12
Method__exit__
(self, *args)
ace_step/cpu_offload.py:17
Method__getitem__
Get item at index with error handling Args: idx: Dataset index Returns: dict: Example featur
ace_step/text2music_dataset.py:653
Method__init__
(self)
ace_step_nodes.py:320
Method__init__
(self, root_dir=DEFAULT_ROOT_DIR)
ace_step/data_sampler.py:11
Method__init__
Initialize the Text2Music dataset Args: train: Whether this is a training dataset train_dataset_path: P
ace_step/text2music_dataset.py:71
Method__init__
(self, model, device="cpu")
ace_step/cpu_offload.py:7
Method__init__
(self, music_dcae, ace_step, umt5encoder, text_tokenizer, device, dtype, overlapped_decode=False, **kwargs)
ace_step/pipeline_ace_step.py:34
Method__init__
(self, momentum: float = -0.75)
ace_step/apg_guidance.py:5
Method__init__
(self, dim, max_position_embeddings=2048, base=10000, device=None)
ace_step/ace_models/ace_step_transformer.py:48
Method__init__
(self, hidden_size, patch_size=[16, 1], out_channels=256)
ace_step/ace_models/ace_step_transformer.py:98
Method__init__
( self, in_channels: Optional[int] = 8, num_layers: int = 28, inner_dim: int =
ace_step/ace_models/ace_step_transformer.py:212
Method__init__
( self, in_features: int, hidden_features: int, out_feature=None, kern
ace_step/ace_models/attention.py:122
Method__init__
( self, dim, num_attention_heads, attention_head_dim, use_adaln_single
ace_step/ace_models/attention.py:190
Method__init__
(self)
ace_step/ace_models/customer_attention_processor.py:29
Method__init__
(self)
ace_step/ace_models/customer_attention_processor.py:209
Method__init__
(self, vocab_file=DEFAULT_VOCAB_FILE)
ace_step/ace_models/lyrics_utils/lyric_tokenizer.py:604
Method__init__
Construct an ConvolutionModule object. Args: channels (int): The number of channels of conv layers. kernel_size (int):
ace_step/ace_models/lyrics_utils/lyric_encoder.py:10
Method__init__
Construct a PositionwiseFeedForward object.
ace_step/ace_models/lyrics_utils/lyric_encoder.py:146
Method__init__
Construct an MultiHeadedAttention object.
ace_step/ace_models/lyrics_utils/lyric_encoder.py:189
Method__init__
Construct an RelPositionMultiHeadedAttention object.
ace_step/ace_models/lyrics_utils/lyric_encoder.py:334
← previousnext →101–200 of 348, ranked by callers