MCPcopy Create free account

hub / github.com/bytedance/DreamID-V / functions

Functions524 in github.com/bytedance/DreamID-V

⨍ Functions 524◇ Types & classes 122
Method__init__
(self, embed_dim=1024, image_size=224, patch_size=14,
dreamidv_wan/modules/clip.py:330
Method__init__
(self, dtype, device, checkpoint_path, tokenizer_path)
dreamidv_wan/modules/clip.py:503
Method__init__
(self, dim, eps=1e-6)
dreamidv_wan/modules/t5.py:55
Method__init__
(self, dim, dim_attn, num_heads, dropout=0.1)
dreamidv_wan/modules/t5.py:71
Method__init__
(self, dim, dim_ffn, dropout=0.1)
dreamidv_wan/modules/t5.py:125
Method__init__
(self, dim, dim_attn, dim_ffn, num_heads,
dreamidv_wan/modules/t5.py:146
Method__init__
(self, dim, dim_attn, dim_ffn, num_heads,
dreamidv_wan/modules/t5.py:180
Method__init__
(self, num_buckets, num_heads, bidirectional, max_dist=128)
dreamidv_wan/modules/t5.py:223
Method__init__
(self, vocab, dim, dim_attn, dim_ffn,
dreamidv_wan/modules/t5.py:269
Method__init__
(self, vocab, dim, dim_attn, dim_ffn,
dreamidv_wan/modules/t5.py:317
Method__init__
( self, text_len, dtype=torch.bfloat16, device=torch.cuda.current_device(),
dreamidv_wan/modules/t5.py:474
Method__init__
(self, *args, **kwargs)
dreamidv_wan/modules/vae.py:22
Method__init__
(self, dim, channel_first=True, images=True, bias=False)
dreamidv_wan/modules/vae.py:41
Method__init__
(self, dim, mode)
dreamidv_wan/modules/vae.py:68
Method__init__
(self, in_dim, out_dim, dropout=0.0)
dreamidv_wan/modules/vae.py:188
Method__init__
(self, dim)
dreamidv_wan/modules/vae.py:228
Method__init__
(self, dim=128, z_dim=4, dim_mult=[1, 2, 4, 4],
dreamidv_wan/modules/vae.py:267
Method__init__
(self, dim=128, z_dim=4, dim_mult=[1, 2, 4, 4],
dreamidv_wan/modules/vae.py:371
Method__init__
(self, dim=128, z_dim=4, dim_mult=[1, 2, 4, 4],
dreamidv_wan/modules/vae.py:485
Method__init__
(self, cin=14, cout=512, vid_dim=1536, patch_size = [1,2,2] )
dreamidv_wan/modules/projector.py:101
Method__init__
(self, dim, num_heads, dropout=0.1, eps=1e-5)
dreamidv_wan/modules/xlm_roberta.py:12
Method__init__
(self, dim, num_heads, post_norm, dropout=0.1, eps=1e-5)
dreamidv_wan/modules/xlm_roberta.py:51
Method__init__
(self, dim, eps=1e-5)
dreamidv_wan/modules/model.py:73
Method__init__
(self, dim, eps=1e-6, elementwise_affine=False)
dreamidv_wan/modules/model.py:92
Method__init__
(self, dim, num_heads, window_size=(-1, -1),
dreamidv_wan/modules/model.py:104
Method__init__
(self, dim, num_heads, window_size=(-1, -1),
dreamidv_wan/modules/model.py:199
Method__init__
(self, cross_attn_type, dim, ffn_dim, num_
dreamidv_wan/modules/model.py:249
Method__init__
(self, in_dim, out_dim)
dreamidv_wan/modules/model.py:362
Method__init__
r""" Initialize the diffusion model backbone. Args: model_type (`str`, *optional*, defaults to 't2v'): Mo
dreamidv_wan/modules/model.py:386
Method__init__
Args: api_key: The API key for Dash Scope authentication and access to related services. model_name: Model name, 'qwe
dreamidv_wan/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
dreamidv_wan/utils/prompt_extend.py:309
Method__init__
(self, pattern: str, **kwargs)
dreamidv_wan/utils/na_resize.py:26
Method__init__
(self, factor)
dreamidv_wan/utils/na_resize.py:34
Method__init__
( self, max_area: float, downsample_only: bool = False, interpolation: Interpo
dreamidv_wan/utils/na_resize.py:56
Method__init__
( self, num_train_timesteps: int = 1000, solver_order: int = 2, prediction_typ
dreamidv_wan/utils/fm_solvers.py:129
Method__init__
( self, num_train_timesteps: int = 1000, solver_order: int = 2,
dreamidv_wan/utils/fm_solvers_unipc.py:77
Method__init__
( self, conditioning_embedding_channels: int, conditioning_channels: int = 3,
express_adaption/poseguider.py:24
Method__init__
(self, FPS=25)
express_adaption/media_pipe/mp_utils.py:32
Method__init__
(self)
express_adaption/media_pipe/__init__.py:27
Method__init__
(self)
express_adaption/media_pipe/__init__.py:108
Method__init__
(self)
express_adaption/media_pipe/__init__.py:194
Method__init__
(self, forehead_edge=False, iris_edge=False, iris_point=False)
express_adaption/media_pipe/draw_util.py:19
Method__init__
(self, cfg, device=None)
pose/extract.py:231
Method__init__
(self, onnx_det, onnx_pose, device = 'cuda:0')
pose/dwpose/wholebody.py:41
Method__len__
(self)
dreamidv_wan_faster/utils/fm_solvers_unipc.py:799
Method__len__
(self)
dreamidv_wan/utils/fm_solvers_unipc.py:799
Function_read_video_decord
read video using decord.VideoReader Args: ele (dict): a dict contains the configuration of video. support keys: - vid
dreamidv_wan_faster/utils/qwen_vl_utils.py:221
Function_read_video_decord
read video using decord.VideoReader Args: ele (dict): a dict contains the configuration of video. support keys: - vid
dreamidv_wan/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:
dreamidv_wan_faster/utils/qwen_vl_utils.py:177
Function_read_video_torchvision
read video using torchvision.io.read_video Args: ele (dict): a dict contains the configuration of video. support keys:
dreamidv_wan/utils/qwen_vl_utils.py:177
Method_sigma_to_t
(self, sigma)
dreamidv_wan_faster/utils/fm_solvers.py:330
Method_sigma_to_t
(self, sigma)
dreamidv_wan_faster/utils/fm_solvers_unipc.py:269
Method_sigma_to_t
(self, sigma)
dreamidv_wan/utils/fm_solvers.py:330
Method_sigma_to_t
(self, sigma)
dreamidv_wan/utils/fm_solvers_unipc.py:269
Functiona
(e,t)
assets/bootstrap.min.js:6
Methodadd_custom_field
(self, key: str, value)
dreamidv_wan_faster/utils/prompt_extend.py:108
Methodadd_custom_field
(self, key: str, value)
dreamidv_wan/utils/prompt_extend.py:108
Methodadd_noise
( self, original_samples: torch.Tensor, noise: torch.Tensor, timesteps: torch.
dreamidv_wan_faster/utils/fm_solvers.py:815
Methodadd_noise
( self, original_samples: torch.Tensor, noise: torch.Tensor, timesteps: torch.
dreamidv_wan_faster/utils/fm_solvers_unipc.py:758
Methodadd_noise
( self, original_samples: torch.Tensor, noise: torch.Tensor, timesteps: torch.
dreamidv_wan/utils/fm_solvers.py:815
Methodadd_noise
( self, original_samples: torch.Tensor, noise: torch.Tensor, timesteps: torch.
dreamidv_wan/utils/fm_solvers_unipc.py:758
Functionattention
( q, k, v, q_lens=None, k_lens=None, dropout_p=0., softmax_scale=None, q_scale
dreamidv_wan_faster/modules/attention.py:133
Functionattention
( q, k, v, q_lens=None, k_lens=None, dropout_p=0., softmax_scale=None, q_scale
dreamidv_wan/modules/attention.py:133
Methodbegin_index
The index for the first timestep. It should be set from pipeline with `set_begin_index` method.
dreamidv_wan_faster/utils/fm_solvers.py:209
Methodbegin_index
The index for the first timestep. It should be set from pipeline with `set_begin_index` method.
dreamidv_wan_faster/utils/fm_solvers_unipc.py:142
Methodbegin_index
The index for the first timestep. It should be set from pipeline with `set_begin_index` method.
dreamidv_wan/utils/fm_solvers.py:209
Methodbegin_index
The index for the first timestep. It should be set from pipeline with `set_begin_index` method.
dreamidv_wan/utils/fm_solvers_unipc.py:142
Functioncache_image
(tensor, save_file, nrow=8, normalize=True, va
dreamidv_wan_faster/utils/utils.py:64
Functioncache_image
(tensor, save_file, nrow=8, normalize=True, va
dreamidv_wan/utils/utils.py:64
Functioncache_video
(tensor, save_file=None, fps=30, suffix='.mp4',
dreamidv_wan_faster/utils/utils.py:23
Methodcross_attn_ffn
(x, context, context_lens, e)
dreamidv_wan_faster/modules/model.py:302
Methodcross_attn_ffn
(x, context, context_lens, e)
dreamidv_wan/modules/model.py:319
Functiondraw_bodypose
(canvas, candidate, subset)
pose/dwpose/util.py:70
Functiondraw_facepose
(canvas, all_lmks)
pose/dwpose/util.py:142
Functiondraw_handpose
(canvas, all_hand_peaks)
pose/dwpose/util.py:114
Methoddraw_mask
(self, image_size, keypoints, normed=False)
express_adaption/media_pipe/draw_util.py:173
Methoddraw_pupils
We have a custom function to draw the pupils because the mp.draw_landmarks method requires a parameter for all landmarks. Until our PR is mer
express_adaption/media_pipe/draw_util.py:108
Methodextend
(self, prompt, system_prompt, seed=-1, *args, **kwargs)
dreamidv_wan_faster/utils/prompt_extend.py:196
Methodextend
(self, prompt, system_prompt, seed=-1, *args, **kwargs)
dreamidv_wan_faster/utils/prompt_extend.py:366
Methodextend
(self, prompt, system_prompt, seed=-1, *args, **kwargs)
dreamidv_wan/utils/prompt_extend.py:196
Methodextend
(self, prompt, system_prompt, seed=-1, *args, **kwargs)
dreamidv_wan/utils/prompt_extend.py:366
Methodextend_with_img
(self, prompt, system_prompt, image: U
dreamidv_wan_faster/utils/prompt_extend.py:232
Methodextend_with_img
(self, prompt, system_prompt, image: U
dreamidv_wan_faster/utils/prompt_extend.py:397
Methodextend_with_img
(self, prompt, system_prompt, image: U
dreamidv_wan/utils/prompt_extend.py:232
Methodextend_with_img
(self, prompt, system_prompt, image: U
dreamidv_wan/utils/prompt_extend.py:397
FunctionfaceDetect
(candidate, subset, oriImg)
pose/dwpose/util.py:223
Methodforward
(self, x)
dreamidv_wan_faster/modules/clip.py:43
Methodforward
(self, x)
dreamidv_wan_faster/modules/clip.py:49
Methodforward
x: [B, L, C].
dreamidv_wan_faster/modules/clip.py:74
Methodforward
(self, x)
dreamidv_wan_faster/modules/clip.py:146
Methodforward
x: [B, L, C].
dreamidv_wan_faster/modules/clip.py:186
Methodforward
(self, x, interpolation=False, use_31_block=False)
dreamidv_wan_faster/modules/clip.py:279
Methodforward
(self, ids)
dreamidv_wan_faster/modules/clip.py:315
Methodforward
imgs: [B, 3, H, W] of torch.float32. - mean: [0.48145466, 0.4578275, 0.40821073] - std: [0.26862954, 0.2613025
dreamidv_wan_faster/modules/clip.py:406
Methodforward
(self, x)
dreamidv_wan_faster/modules/t5.py:48
Methodforward
(self, x)
dreamidv_wan_faster/modules/t5.py:61
Methodforward
x: [B, L1, C]. context: [B, L2, C] or None. mask: [B, L2] or [B, L1, L2] or None.
dreamidv_wan_faster/modules/t5.py:86
Methodforward
(self, x)
dreamidv_wan_faster/modules/t5.py:136
Methodforward
(self, x, mask=None, pos_bias=None)
dreamidv_wan_faster/modules/t5.py:170
Methodforward
(self, x, mask=None, encoder_states=None, enco
dreamidv_wan_faster/modules/t5.py:206
← previousnext →301–400 of 524, ranked by callers