MCPcopy Create free account

hub / github.com/ace-step/ACE-Step / functions

Functions375 in github.com/ace-step/ACE-Step

⨍ Functions 375◇ Types & classes 64↳ Endpoints 4
Method__init__
(self, checkpoint_dir: str, device: str = "cuda")
trainer-api.py:36
Method__init__
( self, learning_rate: float = 1e-4, num_workers: int = 4, train: bool = True,
trainer.py:37
Method__init__
(self, root_dir=DEFAULT_ROOT_DIR)
acestep/data_sampler.py:10
Method__init__
Initialize the Text2Music dataset Args: train: Whether this is a training dataset train_dataset_path: Path t
acestep/text2music_dataset.py:71
Method__init__
(self, model, device="cpu")
acestep/cpu_offload.py:7
Method__init__
( self, checkpoint_dir=None, device_id=0, dtype="bfloat16", text_encod
acestep/pipeline_ace_step.py:95
Method__init__
(self, momentum: float = -0.75)
acestep/apg_guidance.py:5
Method__init__
(self)
acestep/language_segmentation/LangSegment.py:137
Method__init__
(self)
acestep/language_segmentation/LangSegment.py:267
Method__init__
(self, normalized_shape, eps=1e-6, data_format="channels_last")
acestep/music_dcae/music_vocoder.py:80
Method__init__
( self, dim: int, drop_path: float = 0.0, layer_scale_init_value: float = 1e-6
acestep/music_dcae/music_vocoder.py:118
Method__init__
(self, kernel_sizes: List[int], *args, **kwargs)
acestep/music_dcae/music_vocoder.py:172
Method__init__
( self, input_channels=3, depths=[3, 3, 9, 3], dims=[96, 192, 384, 768],
acestep/music_dcae/music_vocoder.py:189
Method__init__
(self, channels, kernel_size=3, dilation=(1, 3, 5))
acestep/music_dcae/music_vocoder.py:277
Method__init__
( self, *, hop_length: int = 512, upsample_rates: Tuple[int] = (8, 8, 2, 2, 2)
acestep/music_dcae/music_vocoder.py:369
Method__init__
( self, input_channels: int = 128, depths: List[int] = [3, 3, 9, 3], dims: Lis
acestep/music_dcae/music_vocoder.py:495
Method__init__
( self, sample_rate=44100, n_fft=2048, win_length=2048, hop_length=512
acestep/music_dcae/music_log_mel.py:68
Method__init__
( self, source_sample_rate=None, dcae_checkpoint_path=DEFAULT_PRETRAINED_PATH,
acestep/music_dcae/music_dcae_pipeline.py:32
Method__init__
( self, num_train_timesteps: int = 1000, shift: float = 1.0, use_dynamic_shift
acestep/schedulers/scheduling_flow_match_euler_discrete.py:65
Method__init__
( self, num_train_timesteps: int = 1000, shift: float = 1.0, use_dynamic_shift
acestep/schedulers/scheduling_flow_match_pingpong.py:65
Method__init__
( self, num_train_timesteps: int = 1000, shift: float = 1.0, sigma_max: Option
acestep/schedulers/scheduling_flow_match_heun_discrete.py:65
Method__init__
(self, dim, max_position_embeddings=2048, base=10000, device=None)
acestep/models/ace_step_transformer.py:48
Method__init__
(self, hidden_size, patch_size=[16, 1], out_channels=256)
acestep/models/ace_step_transformer.py:98
Method__init__
( self, in_channels: Optional[int] = 8, num_layers: int = 28, inner_dim: int =
acestep/models/ace_step_transformer.py:212
Method__init__
( self, in_features: int, hidden_features: int, out_feature=None, kern
acestep/models/attention.py:133
Method__init__
( self, dim, num_attention_heads, attention_head_dim, use_adaln_single
acestep/models/attention.py:202
Method__init__
(self)
acestep/models/customer_attention_processor.py:29
Method__init__
(self)
acestep/models/customer_attention_processor.py:252
Method__init__
(self, vocab_file=DEFAULT_VOCAB_FILE)
acestep/models/lyrics_utils/lyric_tokenizer.py:626
Method__init__
Construct an ConvolutionModule object. Args: channels (int): The number of channels of conv layers. kernel_size (int):
acestep/models/lyrics_utils/lyric_encoder.py:10
Method__init__
Construct a PositionwiseFeedForward object.
acestep/models/lyrics_utils/lyric_encoder.py:146
Method__init__
Construct an MultiHeadedAttention object.
acestep/models/lyrics_utils/lyric_encoder.py:189
Method__init__
Construct an RelPositionMultiHeadedAttention object.
acestep/models/lyrics_utils/lyric_encoder.py:334
Method__init__
Construct an EncoderLayer object.
acestep/models/lyrics_utils/lyric_encoder.py:606
Method__init__
Construct an PositionalEncoding object.
acestep/models/lyrics_utils/lyric_encoder.py:729
Method__init__
Construct ConformerEncoder Args: input_size to use_dynamic_chunk, see in BaseEncoder positionwise_conv_kernel_size (i
acestep/models/lyrics_utils/lyric_encoder.py:905
Method__init__
(self, simplified, traditional)
acestep/models/lyrics_utils/zh_num2words.py:402
Method__init__
(self, power, simplified, traditional, big_s, big_t)
acestep/models/lyrics_utils/zh_num2words.py:421
Method__init__
(self, value, simplified, traditional, big_s, big_t, alt_s=None, alt_t=None)
acestep/models/lyrics_utils/zh_num2words.py:457
Method__init__
(self, simplified, traditional, symbol, expression=None)
acestep/models/lyrics_utils/zh_num2words.py:478
Method__init__
(self, positive, negative, point)
acestep/models/lyrics_utils/zh_num2words.py:505
Method__init__
(self, cardinal=None, chntext=None)
acestep/models/lyrics_utils/zh_num2words.py:757
Method__init__
(self, digit=None, chntext=None)
acestep/models/lyrics_utils/zh_num2words.py:773
Method__init__
(self, telephone=None, raw_chntext=None, chntext=None)
acestep/models/lyrics_utils/zh_num2words.py:789
Method__init__
(self, fraction=None, chntext=None)
acestep/models/lyrics_utils/zh_num2words.py:818
Method__init__
(self, money=None, chntext=None)
acestep/models/lyrics_utils/zh_num2words.py:894
Method__init__
(self, percentage=None, chntext=None)
acestep/models/lyrics_utils/zh_num2words.py:917
Method__init__
( self, to_banjiao: bool = False, to_upper: bool = False, to_lower: bool = Fal
acestep/models/lyrics_utils/zh_num2words.py:1058
Method__iter__
(self)
acestep/models/lyrics_utils/zh_num2words.py:510
Method__len__
Return the number of batches in the dataset
acestep/text2music_dataset.py:224
Method__len__
(self)
acestep/schedulers/scheduling_flow_match_euler_discrete.py:402
Method__len__
(self)
acestep/schedulers/scheduling_flow_match_pingpong.py:342
Method__len__
(self)
acestep/schedulers/scheduling_flow_match_heun_discrete.py:363
Method__len__
(self)
acestep/models/lyrics_utils/lyric_tokenizer.py:740
Method__repr__
(self)
acestep/models/lyrics_utils/zh_num2words.py:410
Method__str__
(self)
acestep/models/lyrics_utils/zh_num2words.py:427
Method__str__
(self)
acestep/models/lyrics_utils/zh_num2words.py:465
Method_init_weights
(self, m)
acestep/music_dcae/music_vocoder.py:250
Method_is_english_word
(self, word)
acestep/language_segmentation/LangSegment.py:354
Method_process_Russian
(self, words, data)
acestep/language_segmentation/LangSegment.py:705
Method_process_Thai
(self, words, data)
acestep/language_segmentation/LangSegment.py:712
Method_process_english
(self, words, data)
acestep/language_segmentation/LangSegment.py:671
Method_process_korean
(self, words, data)
acestep/language_segmentation/LangSegment.py:719
Method_process_number
Numbers alone cannot accurately identify language. Because numbers are universal in all languages. So it won't be executed
acestep/language_segmentation/LangSegment.py:747
Method_process_pinyin
(self, words, data)
acestep/language_segmentation/LangSegment.py:740
Method_process_quotes
(self, words, data)
acestep/language_segmentation/LangSegment.py:726
Method_process_symbol
(self, words, data)
acestep/language_segmentation/LangSegment.py:660
Function_remove_commas
(m)
acestep/models/lyrics_utils/lyric_tokenizer.py:470
Function_remove_dots
(m)
acestep/models/lyrics_utils/lyric_tokenizer.py:477
Method_restore_number
(self, matche)
acestep/language_segmentation/LangSegment.py:638
Methodbegin_index
The index for the first timestep. It should be set from pipeline with `set_begin_index` method.
acestep/schedulers/scheduling_flow_match_euler_discrete.py:103
Methodbegin_index
The index for the first timestep. It should be set from pipeline with `set_begin_index` method.
acestep/schedulers/scheduling_flow_match_pingpong.py:103
Methodbegin_index
The index for the first timestep. It should be set from pipeline with `set_begin_index` method.
acestep/schedulers/scheduling_flow_match_heun_discrete.py:96
Methodchntext2cardinal
(self)
acestep/models/lyrics_utils/zh_num2words.py:761
Methodchntext2fraction
(self)
acestep/models/lyrics_utils/zh_num2words.py:822
Methodchntext2percentage
(self)
acestep/models/lyrics_utils/zh_num2words.py:921
Methodconfigure_optimizers
(self)
trainer.py:413
Functioncpu_offload
(model_attr: str)
acestep/cpu_offload.py:27
Methodcreate
(cls, i, v)
acestep/models/lyrics_utils/zh_num2words.py:469
Methoddecompress
(self, x: Tensor)
acestep/music_dcae/music_log_mel.py:104
Functiondecorator
(func: Callable[..., T])
acestep/cpu_offload.py:28
Functiondump_func
(*args)
acestep/ui/components.py:84
Functionedit_process_func
( text2music_json_data, edit_input_params_json, ed
acestep/ui/components.py:588
Functionedit_type_change_func
(edit_type)
acestep/ui/components.py:549
Methodenable_forward_chunking
Sets the attention processor to use [feed forward chunking](https://huggingface.co/blog/reformer#2-chunked-feed-forward-layers).
acestep/models/ace_step_transformer.py:330
Methodencode
(self, txt, lang)
acestep/models/lyrics_utils/lyric_tokenizer.py:695
Functionextend_process_func
( text2music_json_data, extend_input_params_json,
acestep/ui/components.py:740
Methodextra_repr
(self)
acestep/music_dcae/music_vocoder.py:69
Methodfn_recursive_feed_forward
( module: torch.nn.Module, chunk_size: int, dim: int )
acestep/models/ace_step_transformer.py:351
Methodforward
(self, x)
acestep/music_dcae/music_vocoder.py:66
Methodforward
(self, x)
acestep/music_dcae/music_vocoder.py:90
Methodforward
(self, x, apply_residual: bool = True)
acestep/music_dcae/music_vocoder.py:149
Methodforward
(self, x: torch.Tensor)
acestep/music_dcae/music_vocoder.py:181
Methodforward
( self, x: torch.Tensor, )
acestep/music_dcae/music_vocoder.py:255
Methodforward
(self, x)
acestep/music_dcae/music_vocoder.py:352
Methodforward
(self, x, template=None)
acestep/music_dcae/music_vocoder.py:457
Methodforward
(self, mel)
acestep/music_dcae/music_vocoder.py:568
Methodforward
(self, y: Tensor)
acestep/music_dcae/music_log_mel.py:34
Methodforward
(self, x: Tensor, return_linear: bool = False)
acestep/music_dcae/music_log_mel.py:107
Methodforward
(self, audios, audio_lengths=None, sr=None)
acestep/music_dcae/music_dcae_pipeline.py:355
← previousnext →201–300 of 375, ranked by callers