MCPcopy Create free account

hub / github.com/character-ai/Ovi / functions

Functions478 in github.com/character-ai/Ovi

Method__call__
(self, texts, device)
ovi/modules/t5.py:507
Method__call__
(self, prompt, tar_lang="ch", image=None,
ovi/utils/prompt_extend.py:138
Method__init__
(self, config=DEFAULT_CONFIG, device=0, target_dtype=torch.bfloat16)
ovi/ovi_fusion_engine.py:52
Method__init__
(self, name, seq_len=None, clean=None, **kwargs)
ovi/modules/tokenizers.py:39
Method__init__
(self, dim, num_heads, causal=False, attn_
ovi/modules/clip.py:55
Method__init__
(self, dim, mlp_ratio, num_heads, post_nor
ovi/modules/clip.py:115
Method__init__
(self, dim, mlp_ratio, num_heads, activati
ovi/modules/clip.py:159
Method__init__
(self, image_size=224, patch_size=16, dim=768,
ovi/modules/clip.py:214
Method__init__
(self, **kwargs)
ovi/modules/clip.py:308
Method__init__
(self, embed_dim=1024, image_size=224, patch_size=14,
ovi/modules/clip.py:333
Method__init__
(self, dtype, device, checkpoint_path, tokenizer_path)
ovi/modules/clip.py:506
Method__init__
(self, dim, eps=1e-6)
ovi/modules/t5.py:55
Method__init__
(self, dim, dim_attn, num_heads, dropout=0.1)
ovi/modules/t5.py:71
Method__init__
(self, dim, dim_ffn, dropout=0.1)
ovi/modules/t5.py:125
Method__init__
(self, dim, dim_attn, dim_ffn, num_heads,
ovi/modules/t5.py:146
Method__init__
(self, dim, dim_attn, dim_ffn, num_heads,
ovi/modules/t5.py:180
Method__init__
(self, num_buckets, num_heads, bidirectional, max_dist=128)
ovi/modules/t5.py:223
Method__init__
(self, vocab, dim, dim_attn, dim_ffn,
ovi/modules/t5.py:269
Method__init__
(self, vocab, dim, dim_attn, dim_ffn,
ovi/modules/t5.py:317
Method__init__
( self, text_len, dtype=torch.bfloat16, device=torch.cuda.current_device(),
ovi/modules/t5.py:474
Method__init__
(self, *args, **kwargs)
ovi/modules/vae.py:22
Method__init__
(self, dim, channel_first=True, images=True, bias=False)
ovi/modules/vae.py:41
Method__init__
(self, dim, mode)
ovi/modules/vae.py:68
Method__init__
(self, in_dim, out_dim, dropout=0.0)
ovi/modules/vae.py:188
Method__init__
(self, dim)
ovi/modules/vae.py:228
Method__init__
(self, dim=128, z_dim=4, dim_mult=[1, 2, 4, 4],
ovi/modules/vae.py:267
Method__init__
(self, dim=128, z_dim=4, dim_mult=[1, 2, 4, 4],
ovi/modules/vae.py:371
Method__init__
(self, dim=128, z_dim=4, dim_mult=[1, 2, 4, 4],
ovi/modules/vae.py:485
Method__init__
(self, *args, **kwargs)
ovi/modules/vae2_2.py:22
Method__init__
(self, dim, channel_first=True, images=True, bias=False)
ovi/modules/vae2_2.py:47
Method__init__
(self, dim, mode)
ovi/modules/vae2_2.py:73
Method__init__
(self, in_dim, out_dim, dropout=0.0)
ovi/modules/vae2_2.py:195
Method__init__
(self, dim)
ovi/modules/vae2_2.py:243
Method__init__
( self, in_channels, out_channels, factor_t, factor_s=1, )
ovi/modules/vae2_2.py:318
Method__init__
(self, in_dim, out_dim, dropout, mult,
ovi/modules/vae2_2.py:417
Method__init__
(self, in_dim, out_dim, dropout, mult,
ovi/modules/vae2_2.py:457
Method__init__
( self, dim=128, z_dim=4, dim_mult=[1, 2, 4, 4], num_res_blocks=2,
ovi/modules/vae2_2.py:502
Method__init__
( self, dim=128, z_dim=4, dim_mult=[1, 2, 4, 4], num_res_blocks=2,
ovi/modules/vae2_2.py:618
Method__init__
( self, dim=160, dec_dim=256, z_dim=16, dim_mult=[1, 2, 4, 4],
ovi/modules/vae2_2.py:736
Method__init__
( self, z_dim=48, c_dim=160, vae_pth=None, dim_mult=[1, 2, 4, 4],
ovi/modules/vae2_2.py:890
Method__init__
(self, video_config=None, audio_config=None)
ovi/modules/fusion.py:10
Method__init__
(self, dim, num_heads, dropout=0.1, eps=1e-5)
ovi/modules/xlm_roberta.py:12
Method__init__
(self, dim, num_heads, post_norm, dropout=0.1, eps=1e-5)
ovi/modules/xlm_roberta.py:51
Method__init__
Initialize the FeedForward module. Args: dim (int): Input dimension. hidden_dim (int): Hidden dimension of t
ovi/modules/model.py:121
Method__init__
(self, dim, eps=1e-5)
ovi/modules/model.py:168
Method__init__
(self, dim, eps=1e-6, elementwise_affine=False)
ovi/modules/model.py:187
Method__init__
(self, dim, num_heads, window_size=(-1, -1),
ovi/modules/model.py:200
Method__init__
(self, dim, num_heads, window_size=(-1, -1),
ovi/modules/model.py:299
Method__init__
(self, dim, num)
ovi/modules/model.py:369
Method__init__
(self, cross_attn_type, dim, ffn_dim, num_
ovi/modules/model.py:378
Method__init__
(self, in_dim, out_dim)
ovi/modules/model.py:507
Method__init__
r""" Initialize the diffusion model backbone. Args: model_type (`str`, *optional*, defaults to 't2v'): Mo
ovi/modules/model.py:531
Method__init__
( self, *, tod_vae_ckpt: str, bigvgan_vocoder_ckpt: Optional[str] = None,
ovi/modules/mmaudio/features_utils.py:35
Method__init__
( self, *, sampling_rate: float = 16_000, n_fft: int = 1024, num_mels:
ovi/modules/mmaudio/ext/stft_converter_mel.py:21
Method__init__
( self, *, sampling_rate: float, n_fft: int, num_mels: int, ho
ovi/modules/mmaudio/ext/mel_converter.py:20
Method__init__
( self, *, sampling_rate: float = 16_000, n_fft: int = 1024, num_mels:
ovi/modules/mmaudio/ext/stft_converter.py:21
Method__init__
(self, *, dim: int, ch_mult: tuple[int] = (1, 2, 4, 8),
ovi/modules/mmaudio/ext/autoencoder/vae.py:172
Method__init__
(self, *, dim: int, out_dim: int, ch_mult:
ovi/modules/mmaudio/ext/autoencoder/vae.py:268
Method__init__
(self, *, in_dim, out_dim=None, conv_shortcut=False, kernel_size=3, use_norm=True)
ovi/modules/mmaudio/ext/autoencoder/vae_modules.py:16
Method__init__
(self, in_channels, num_heads=1)
ovi/modules/mmaudio/ext/autoencoder/vae_modules.py:56
Method__init__
(self, in_channels, with_conv)
ovi/modules/mmaudio/ext/autoencoder/vae_modules.py:99
Method__init__
(self, *, vae_ckpt_path, vocoder_ckpt_path: Optional[str] =
ovi/modules/mmaudio/ext/autoencoder/autoencoder.py:17
Method__init__
(self, dim=None, eps=1e-4)
ovi/modules/mmaudio/ext/autoencoder/edm2_utils.py:65
Method__init__
(self, parameters, deterministic=False)
ovi/modules/mmaudio/ext/autoencoder/distributions.py:8
Method__init__
( self, h: AttrDict, channels: int, kernel_size: int = 3,
ovi/modules/mmaudio/ext/bigvgan_v2/bigvgan.py:45
Method__init__
( self, h: AttrDict, channels: int, kernel_size: int = 3,
ovi/modules/mmaudio/ext/bigvgan_v2/bigvgan.py:143
Method__init__
Initialization. INPUT: - in_features: shape of the input - alpha - trainable parameter that controls frequenc
ovi/modules/mmaudio/ext/bigvgan_v2/activations.py:83
Method__init__
(self, *args, **kwargs)
ovi/modules/mmaudio/ext/bigvgan_v2/env.py:9
Method__init__
(self, ratio=2, kernel_size=None)
ovi/modules/mmaudio/ext/bigvgan_v2/alias_free_activation/torch/resample.py:39
Method__init__
kernel_size should be even number for stylegan3 setup, in this implementation, odd number is also possible.
ovi/modules/mmaudio/ext/bigvgan_v2/alias_free_activation/torch/filter.py:66
Method__init__
( self, activation, up_ratio: int = 2, down_ratio: int = 2, up_kernel_
ovi/modules/mmaudio/ext/bigvgan_v2/alias_free_activation/torch/act.py:11
Method__init__
( self, activation, up_ratio: int = 2, down_ratio: int = 2, up_kernel_
ovi/modules/mmaudio/ext/bigvgan_v2/alias_free_activation/cuda/activation1d.py:36
Method__init__
(self, h, channels, kernel_size=3, dilation=(1, 3), activation=None)
ovi/modules/mmaudio/ext/bigvgan/models.py:115
Method__init__
(self, h)
ovi/modules/mmaudio/ext/bigvgan/models.py:171
Method__init__
(self, ckpt_path, config_path=_bigvgan_vocoder_path)
ovi/modules/mmaudio/ext/bigvgan/bigvgan.py:14
Method__init__
Initialization. INPUT: - in_features: shape of the input - alpha - trainable parameter that controls frequenc
ovi/modules/mmaudio/ext/bigvgan/activations.py:79
Method__init__
(self, *args, **kwargs)
ovi/modules/mmaudio/ext/bigvgan/env.py:9
Method__init__
(self, ratio=2, kernel_size=None)
ovi/modules/mmaudio/ext/bigvgan/alias_free_torch/resample.py:37
Method__init__
(self, cutoff=0.5, half_width=0.6, stride: int = 1,
ovi/modules/mmaudio/ext/bigvgan/alias_free_torch/filter.py:61
Method__init__
(self, activation, up_ratio: int = 2, down_ratio: int = 2,
ovi/modules/mmaudio/ext/bigvgan/alias_free_torch/act.py:9
Method__init__
Args: api_key: The API key for Dash Scope authentication and access to related services. model_name: Model name, 'qwe
ovi/utils/prompt_extend.py:159
Method__init__
Args: model_name: Use predefined model names such as 'QwenVL2.5_7B' and 'Qwen2.5_14B', which are specific version
ovi/utils/prompt_extend.py:309
Method__init__
( self, num_train_timesteps: int = 1000, solver_order: int = 2, prediction_typ
ovi/utils/fm_solvers.py:129
Method__init__
( self, num_train_timesteps: int = 1000, solver_order: int = 2,
ovi/utils/fm_solvers_unipc.py:77
Method__init__
(self)
ovi/distributed_comms/parallel_states.py:8
Method__len__
(self)
ovi/utils/fm_solvers.py:859
Method__len__
(self)
ovi/utils/fm_solvers_unipc.py:799
Method_from_pretrained
Load Pytorch pretrained weights and return the loaded model.
ovi/modules/mmaudio/ext/bigvgan_v2/bigvgan.py:361
Function_read_video_decord
read video using decord.VideoReader Args: ele (dict): a dict contains the configuration of video. support keys: - vid
ovi/utils/qwen_vl_utils.py:221
Function_read_video_torchvision
read video using torchvision.io.read_video Args: ele (dict): a dict contains the configuration of video. support keys:
ovi/utils/qwen_vl_utils.py:177
Method_save_pretrained
Save weights and config.json from a Pytorch model to a local directory.
ovi/modules/mmaudio/ext/bigvgan_v2/bigvgan.py:350
Method_sigma_to_t
(self, sigma)
ovi/utils/fm_solvers.py:330
Method_sigma_to_t
(self, sigma)
ovi/utils/fm_solvers_unipc.py:269
Methodadd_custom_field
(self, key: str, value)
ovi/utils/prompt_extend.py:108
Methodadd_noise
( self, original_samples: torch.Tensor, noise: torch.Tensor, timesteps: torch.
ovi/utils/fm_solvers.py:815
Methodadd_noise
( self, original_samples: torch.Tensor, noise: torch.Tensor, timesteps: torch.
ovi/utils/fm_solvers_unipc.py:758
Functionapply_rope
(x: Tensor, rot: Tensor)
ovi/modules/mmaudio/ext/rotary_embeddings.py:30
Functionapply_weight_norm
(m)
ovi/modules/mmaudio/ext/bigvgan_v2/utils.py:16
Functionapply_weight_norm
(m)
ovi/modules/mmaudio/ext/bigvgan/utils.py:16
Functionattention
( q, k, v, q_lens=None, k_lens=None, dropout_p=0., softmax_scale=None, q_scale
ovi/modules/attention.py:250
← previousnext →201–300 of 478, ranked by callers