MCPcopy Create free account

hub / github.com/Francis-Rings/FlashPortrait / functions

Functions508 in github.com/Francis-Rings/FlashPortrait

↓ 1 callersMethodprepare_latents
( self, batch_size, num_channels_latents, num_frames, height, width, dtype, device, generator, latents
wan/pipeline/pipeline_wan_fun_inpaint.py:314
↓ 1 callersMethodprepare_latents
( self, batch_size, num_channels_latents, num_frames, height, width, dtype, device, generator, latents
wan/pipeline/pipeline_wan_fun_control.py:315
↓ 1 callersMethodprepare_latents
( self, batch_size, num_channels_latents, num_frames, height, width, dtype, device, generator, latents
wan/pipeline/pipeline_wan.py:264
↓ 1 callersMethodprepare_latents
( self, batch_size, num_channels_latents, num_frames, height, width, dtype, device, generator, la
wan/pipeline/pipeline_wan_long.py:360
↓ 1 callersMethodprepare_mask_latents
( self, mask, masked_image, batch_size, height, width, dtype, device, generator, do_classifier_free_gu
wan/pipeline/pipeline_wan_fun_inpaint.py:341
↓ 1 callersMethodprepare_mask_latents
( self, mask, masked_image, batch_size, height, width, dtype, device, generator, do_classifier_fr
wan/pipeline/pipeline_wan_long.py:387
↓ 1 callersFunctionremove_step_distill_lora
Remove step distillation LoRA from the pipeline's transformer. This function subtracts the LoRA weights from the transformer, restor
wan/utils/step_distill_lora.py:214
↓ 1 callersMethodreparameterize
(self, mu, log_var)
wan/models/wan_vae.py:580
↓ 1 callersMethodreset
(self)
wan/models/cache_utils.py:73
↓ 1 callersFunctionresize_mask
(mask, latent, process_first_frame_only=True)
train_portrait.py:116
↓ 1 callersFunctionresize_mask
(mask, latent, process_first_frame_only=True)
wan/pipeline/pipeline_wan_fun_inpaint.py:99
↓ 1 callersFunctionresize_mask
(mask, latent, process_first_frame_only=True)
wan/pipeline/pipeline_wan_long.py:137
↓ 1 callersFunctionrope_apply_qk
(q, k, grid_sizes, freqs)
wan/models/wan_transformer3d.py:167
↓ 1 callersFunctionsetup_tiled_vae
Wrap VAE with tiling support for high resolution processing This implementation matches LightX2V's tiled_encode/tiled_decode exactly:
wan/models/wan_vae_tiled.py:307
↓ 1 callersFunctionsetup_tiny_vae
Replace pipeline VAE with tiny VAE for faster decoding. Args: pipeline: WanI2VLongPipeline instance model_type: "wan2.1"
wan/models/wan_vae_tiny_pipeline.py:11
↓ 1 callersMethodsmooth
(self, new_points, face_dis)
wan/models/face_model.py:70
↓ 1 callersMethodstep
Predict the sample from the previous timestep by reversing the SDE. This function propagates the sample with the multistep UniPC.
wan/utils/fm_solvers_unipc.py:654
↓ 1 callersMethodtiled_decode
Decode latents with tiling for high resolution (matches LightX2V exactly) Key differences from simple implementation:
wan/models/wan_vae_tiled.py:189
↓ 1 callersMethodtiled_encode
Encode video with tiling for high resolution (matches LightX2V exactly) Key differences from simple implementation:
wan/models/wan_vae_tiled.py:102
↓ 1 callersMethodtime_shift
(self, mu: float, sigma: float, t: torch.Tensor)
wan/utils/fm_solvers.py:337
↓ 1 callersMethodtime_shift
(self, mu: float, sigma: float, t: torch.Tensor)
wan/utils/fm_solvers_unipc.py:275
↓ 1 callersFunctionumeyama
Estimate N-D similarity transformation with or without scaling. Parameters ---------- src : (M, N) array Source coordinates.
wan/models/pdf.py:21
↓ 1 callersFunctionunpatchify
(x, patch_size)
wan/models/wan_vae3_8.py:304
↓ 1 callersMethodunpatchify
r""" Reconstruct video tensors from patch embeddings. Args: x (List[Tensor]): List of patchified features
wan/models/wan_transformer3d.py:1166
↓ 1 callersFunctionunwrap_model
(model)
train_portrait.py:1136
↓ 1 callersFunctionwarp_boxes
Apply transform to boxes Copy from nanodet/data/transform/warp.py
wan/models/face_det.py:56
FunctionVideoReader_contextmanager
(*args, **kwargs)
wan/data/utils.py:115
Method__call__
(self, n_samples, generator=None, device=None)
wan/utils/discrete_sampler.py:37
Method__call__
(self, x, x_prev, fcmin=1.0, min_cutoff=1.0, beta=0.0)
wan/models/face_model.py:21
Method__call__
(self, x, x_prev, fcmin=1.0, min_cutoff=1.0, beta=0.0)
wan/models/face_utils.py:58
Method__call__
Function invoked when calling the pipeline for generation. Args: Examples: Returns:
wan/pipeline/pipeline_wan_fun_inpaint.py:471
Method__call__
Function invoked when calling the pipeline for generation. Args: Examples: Returns:
wan/pipeline/pipeline_wan_fun_control.py:470
Method__call__
Function invoked when calling the pipeline for generation. Args: Examples: Returns:
wan/pipeline/pipeline_wan.py:386
Method__call__
Function invoked when calling the pipeline for generation. Args: Examples: Returns:
wan/pipeline/pipeline_wan_long.py:518
Method__getattr__
(self, name)
wan/models/wan_vae_tiled.py:296
Method__getitem__
(self, idx)
wan/data/portrait_data.py:141
Method__init__
(self, transformer3d, portrait_encoder)
train_portrait.py:175
Method__init__
(self, num_idx, uniform_sampling=False, start_num_idx=0, sp_size=1)
wan/utils/discrete_sampler.py:6
Method__init__
( self, num_train_timesteps: int = 1000, shift: float = 5.0, denoising_step_li
wan/utils/step_distill_scheduler.py:41
Method__init__
Args: original_scheduler: The original scheduler from the pipeline denoising_step_list: List of timesteps for step di
wan/utils/step_distill_scheduler.py:238
Method__init__
( self, num_train_timesteps: int = 1000, solver_order: int = 2, prediction_typ
wan/utils/fm_solvers.py:129
Method__init__
( self, text_encoder: Union[List[T5EncoderModel], T5EncoderModel], unet, multi
wan/utils/lora_utils.py:167
Method__init__
( self, num_train_timesteps: int = 1000, solver_order: int = 2,
wan/utils/fm_solvers_unipc.py:77
Method__init__
(self, entry)
wan/data/utils.py:206
Method__init__
(self, txt_path, width, height, n_sample_frames, sample_frame_rate, enable_inpaint=True, face_aligner=None)
wan/data/portrait_data.py:104
Method__init__
(self, gpu_id=None, alignment_model_path="", det_model_path="")
wan/models/face_align.py:12
Method__init__
(self, sample)
wan/models/wan_vae_tiny_pipeline.py:7
Method__init__
(self, dx0=0.0, d_cutoff=1.0)
wan/models/face_model.py:17
Method__init__
(self)
wan/models/face_model.py:44
Method__init__
(self, face_alignment_module, reset=False)
wan/models/face_model.py:328
Method__init__
( self, vae, tile_sample_min_height=256, tile_sample_min_width=256, ti
wan/models/wan_vae_tiled.py:45
Method__init__
(self, num_modules, depth, num_features)
wan/models/pdf.py:253
Method__init__
(self, in_planes, out_planes)
wan/models/pdf.py:302
Method__init__
(self, pose_dim=6, eye_dim=6)
wan/models/pdf.py:412
Method__init__
(self, *, dim, dim_head=64, heads=8)
wan/models/portrait_encoder.py:28
Method__init__
(self, adapter_in_dim=1024, cross_attention_dim=1024)
wan/models/portrait_encoder.py:122
Method__init__
(self, adapter_in_dim: int, adapter_proj_dim: int)
wan/models/portrait_encoder.py:151
Method__init__
(self, dim)
wan/models/wan_camera_adapter.py:45
Method__init__
(self, dim, eps=1e-5)
wan/models/wan_transformer3d.py:175
Method__init__
(self, dim, eps=1e-6, elementwise_affine=False)
wan/models/wan_transformer3d.py:194
Method__init__
(self, dim, num_heads, window_size=(-1, -1),
wan/models/wan_transformer3d.py:207
Method__init__
(self, dim, num_heads, window_size=(-1, -1),
wan/models/wan_transformer3d.py:298
Method__init__
(self, cross_attn_type, dim, ffn_dim, num_
wan/models/wan_transformer3d.py:436
Method__init__
(self, in_dim, out_dim)
wan/models/wan_transformer3d.py:590
Method__init__
r""" Initialize the diffusion model backbone. Args: model_type (`str`, *optional*, defaults to 't2v'): Mo
wan/models/wan_transformer3d.py:616
Method__init__
(self, dim, num_heads, dropout=0.1, eps=1e-5)
wan/models/wan_xlm_roberta.py:12
Method__init__
(self, dim, num_heads, post_norm, dropout=0.1, eps=1e-5)
wan/models/wan_xlm_roberta.py:51
Method__init__
(self, dim, dim_attn, num_heads, dropout=0.1)
wan/models/wan_text_encoder.py:60
Method__init__
(self, dim, dim_ffn, dropout=0.1)
wan/models/wan_text_encoder.py:114
Method__init__
(self, dim, dim_attn, dim_ffn, num_heads,
wan/models/wan_text_encoder.py:134
Method__init__
(self, dim, dim_attn, dim_ffn, num_heads,
wan/models/wan_text_encoder.py:167
Method__init__
(self, num_buckets, num_heads, bidirectional, max_dist=128)
wan/models/wan_text_encoder.py:209
Method__init__
(self, vocab, dim, dim_attn, dim_ffn,
wan/models/wan_text_encoder.py:257
Method__init__
Initialize the one euro filter.
wan/models/face_utils.py:50
Method__init__
(self, *args, **kwargs)
wan/models/wan_vae.py:26
Method__init__
(self, dim, mode)
wan/models/wan_vae.py:72
Method__init__
(self, in_dim, out_dim, dropout=0.0)
wan/models/wan_vae.py:192
Method__init__
(self, dim)
wan/models/wan_vae.py:232
Method__init__
(self, dim=128, z_dim=4, dim_mult=[1, 2, 4, 4],
wan/models/wan_vae.py:271
Method__init__
(self, dim=128, z_dim=4, dim_mult=[1, 2, 4, 4],
wan/models/wan_vae.py:375
Method__init__
(self, dim=128, z_dim=4, dim_mult=[1, 2, 4, 4],
wan/models/wan_vae.py:489
Method__init__
( self, latent_channels=16, temporal_compression_ratio=4, spatial_compression_
wan/models/wan_vae.py:627
Method__init__
( self, attn_scales = [], base_dim = 96, dim_mult = [ 1,
wan/models/wan_vae.py:725
Method__init__
(self, dim, num_heads, causal=False, attn_
wan/models/wan_image_encoder.py:57
Method__init__
(self, dim, mlp_ratio, num_heads, post_nor
wan/models/wan_image_encoder.py:116
Method__init__
(self, dim, mlp_ratio, num_heads, activati
wan/models/wan_image_encoder.py:160
Method__init__
(self, image_size=224, patch_size=16, dim=768,
wan/models/wan_image_encoder.py:213
Method__init__
(self, **kwargs)
wan/models/wan_image_encoder.py:307
Method__init__
(self, embed_dim=1024, image_size=224, patch_size=14,
wan/models/wan_image_encoder.py:332
Method__init__
(self)
wan/models/wan_image_encoder.py:505
Method__init__
(self, n_in, n_out, act_func)
wan/models/wan_vae_tiny.py:23
Method__init__
(self, n_f, stride)
wan/models/wan_vae_tiny.py:45
Method__init__
(self, checkpoint_path="taehv.pth", decoder_time_upscale=(True, True), decoder_space_upscale=(True, True, True
wan/models/wan_vae_tiny.py:121
Method__init__
(self, vae_path="taew2_1.pth", dtype=torch.bfloat16, device="cuda", need_scaled=False)
wan/models/wan_vae_tiny.py:231
Method__init__
(self, vae_path="taew2_2.pth", dtype=torch.bfloat16, device="cuda", need_scaled=False)
wan/models/wan_vae_tiny.py:274
Method__init__
( self, coefficients: list[float], num_steps: int, rel_l1_thresh: float = 0.0,
wan/models/cache_utils.py:33
Method__init__
( self, input_shape=[272, 160], reg_max=7, strides=[8, 16, 32], p
wan/models/face_det.py:179
Method__init__
(self, model_path="", gpu_id=None, *args, **kwargs)
wan/models/face_det.py:308
Method__init__
(self, *args, **kwargs)
wan/models/wan_vae3_8.py:27
Method__init__
(self, dim, channel_first=True, images=True, bias=False)
wan/models/wan_vae3_8.py:52
← previousnext →201–300 of 508, ranked by callers