MCPcopy Create free account

hub / github.com/OpenImagingLab/FlashVSR / functions

Functions2,110 in github.com/OpenImagingLab/FlashVSR

↓ 1 callersFunctionload_state_dict_from_folder
(file_path, torch_dtype=None)
diffsynth/models/utils.py:55
↓ 1 callersFunctionload_state_dict_from_safetensors
(file_path, torch_dtype=None)
diffsynth/models/utils.py:72
↓ 1 callersMethodload_textual_inversions
(self, model_paths)
diffsynth/prompters/sd_prompter.py:48
↓ 1 callersFunctionlow_version_attention
(query, key, value, attn_bias=None)
diffsynth/models/attention.py:5
↓ 1 callersFunctionmain
()
examples/WanVSR/infer_flashvsr_v1.1_tiny_long_video.py:200
↓ 1 callersFunctionmain
()
examples/WanVSR/infer_flashvsr_v1.1_tiny.py:196
↓ 1 callersFunctionmain
()
examples/WanVSR/infer_flashvsr_full.py:180
↓ 1 callersFunctionmain
()
examples/WanVSR/infer_flashvsr_tiny_long_video.py:200
↓ 1 callersFunctionmain
()
examples/WanVSR/infer_flashvsr_v1.1_full.py:180
↓ 1 callersFunctionmain
()
examples/WanVSR/infer_flashvsr_tiny.py:196
↓ 1 callersMethodmask
(self, height, width, border_width)
diffsynth/models/tiler.py:10
↓ 1 callersMethodmatch_using_filename
(self, file_names_a, file_names_b)
diffsynth/extensions/FastBlend/api.py:121
↓ 1 callersMethodmatch_using_numbers
(self, file_names_a, file_names_b)
diffsynth/extensions/FastBlend/api.py:131
↓ 1 callersMethodmerge_latents
(self, value, latents, masks, scales, blur_kernel_size=33, blur_sigma=10.0)
diffsynth/pipelines/base.py:51
↓ 1 callersFunctionmodel_fn_wan_video
( dit: WanModel, x: torch.Tensor, timestep: torch.Tensor, context: torch.Tensor, tea_cache
diffsynth/pipelines/flashvsr_tiny.py:489
↓ 1 callersFunctionmodel_fn_wan_video
( dit: WanModel, x: torch.Tensor, timestep: torch.Tensor, context: torch.Tensor, tea_cache
diffsynth/pipelines/flashvsr_full.py:506
↓ 1 callersFunctionmodel_fn_wan_video
( dit: WanModel, x: torch.Tensor, timestep: torch.Tensor, context: torch.Tensor, tea_cache
diffsynth/pipelines/flashvsr_tiny_long.py:490
↓ 1 callersFunctionmodulate
(x, shift, scale)
diffsynth/models/omnigen.py:191
↓ 1 callersMethodmodulate
(self, x, shift, scale)
diffsynth/models/hunyuan_dit.py:77
↓ 1 callersMethodnaive_encode
(self, x, is_init_image=True)
diffsynth/models/stepvideo_vae.py:966
↓ 1 callersMethodneighboor_step
(self, nnf, d)
diffsynth/extensions/FastBlend/patch_match.py:100
↓ 1 callersMethodoffload
(self)
diffsynth/vram_management/layers.py:23
↓ 1 callersMethodonload
(self)
diffsynth/vram_management/layers.py:28
↓ 1 callersFunctionpad_freqs
(original_tensor, target_len)
diffsynth/distributed/xdit_context_parallel.py:15
↓ 1 callersMethodpad_input_ids
(self, input_ids, image_sizes)
diffsynth/prompters/omnigen_prompter.py:233
↓ 1 callersFunctionpatch_tensor_constructor
(fn)
diffsynth/models/utils.py:26
↓ 1 callersMethodpatch_tgrow_layers
Patch TGrow layers to use a smaller kernel if needed (decoder-only).
examples/WanVSR/utils/TCDecoder.py:244
↓ 1 callersMethodpatchfy
(self, hidden_states)
diffsynth/models/stepvideo_dit.py:812
↓ 1 callersMethodpatchify
(self, x: torch.Tensor)
diffsynth/models/wan_video_dit.py:580
↓ 1 callersMethodpixel_shuffle_3d
(x: torch.Tensor, factor: int)
diffsynth/models/stepvideo_vae.py:307
↓ 1 callersMethodpixel_unshuffle_3d
(x: torch.Tensor, factor: int)
diffsynth/models/stepvideo_vae.py:344
↓ 1 callersFunctionpos_interpolate
(pos, seq_len)
diffsynth/models/wan_video_image_encoder.py:203
↓ 1 callersMethodpost_process_latents
(self, latents, post_normalize=True, contrast_enhance_scale=1.0)
diffsynth/pipelines/svd_video.py:102
↓ 1 callersFunctionprecompute_freqs_cis_3d
(dim: int, end: int = 1024, theta: float = 10000.0)
diffsynth/models/wan_video_dit.py:250
↓ 1 callersMethodprefetch_layer
Starts prefetching the next layer cache
diffsynth/pipelines/omnigen_image.py:28
↓ 1 callersMethodprefetch_layer
Starts prefetching the next layer cache
diffsynth/models/omnigen.py:27
↓ 1 callersMethodprepare_attn_mask
(self, encoder_attention_mask, encoder_hidden_states, q_seqlen)
diffsynth/models/stepvideo_dit.py:817
↓ 1 callersFunctionprepare_causal_attention_mask
(n_frame, n_hw, dtype, device, batch_size=None)
diffsynth/models/hunyuan_video_vae_decoder.py:107
↓ 1 callersMethodprepare_controlnet
(self, controlnet_image, masks, controlnet_inpaint_mask, tiler_kwargs, enable_controlnet_on_negative)
diffsynth/pipelines/flux_image.py:330
↓ 1 callersMethodprepare_controlnet_kwargs
(self, control_video, num_frames, height, width, clip_feature=None, y=None, tiled=True, tile_size=(34, 34), ti
diffsynth/pipelines/wan_video.py:240
↓ 1 callersMethodprepare_eligen
(self, prompt_emb_nega, eligen_entity_prompts, eligen_entity_masks, width, height, t5_sequence_length, enable_
diffsynth/pipelines/flux_image.py:345
↓ 1 callersMethodprepare_entity_inputs
(self, entity_prompts, entity_masks, width, height, t5_sequence_length=512, enable_eligen_inpaint=False)
diffsynth/pipelines/flux_image.py:290
↓ 1 callersMethodprepare_extra_emb
(self, text_emb_t5, timestep, size_emb, dtype, batch_size)
diffsynth/models/hunyuan_dit.py:271
↓ 1 callersMethodprepare_extra_input
(self, latents=None, guidance=1.0)
diffsynth/pipelines/hunyuan_video.py:112
↓ 1 callersMethodprepare_extra_input
(self, latents)
diffsynth/pipelines/cog_video.py:57
↓ 1 callersMethodprepare_extra_input
(self, latents=None, guidance=1.0)
diffsynth/pipelines/flux_image.py:215
↓ 1 callersMethodprepare_extra_input
(self, latents=None, tiled=False, tile_size=64, tile_stride=32)
diffsynth/pipelines/hunyuan_image.py:193
↓ 1 callersMethodprepare_extra_input
(self, latents=None)
diffsynth/pipelines/sdxl_image.py:105
↓ 1 callersMethodprepare_extra_input
(self, latents=None)
diffsynth/pipelines/wan_video.py:259
↓ 1 callersMethodprepare_flex_kwargs
(self, latents, flex_inpaint_image=None, flex_inpaint_mask=None, flex_control_image=None, flex_control_strengt
diffsynth/pipelines/flux_image.py:380
↓ 1 callersMethodprepare_freqs
(self, latents)
diffsynth/models/hunyuan_video_dit.py:628
↓ 1 callersMethodprepare_image_ids
(self, latents)
diffsynth/models/flux_controlnet.py:29
↓ 1 callersMethodprepare_infinite_you
(self, id_image, controlnet_image, infinityou_guidance, height, width)
diffsynth/pipelines/flux_image.py:373
↓ 1 callersMethodprepare_infinite_you
(self, model, id_image, controlnet_image, infinityou_guidance, height, width)
diffsynth/pipelines/flux_image.py:595
↓ 1 callersFunctionprepare_input_tensor
(path: str, scale: float = 4, dtype=torch.bfloat16, device='cuda')
examples/WanVSR/infer_flashvsr_v1.1_tiny_long_video.py:87
↓ 1 callersFunctionprepare_input_tensor
(path: str, scale: float = 4, dtype=torch.bfloat16, device='cuda')
examples/WanVSR/infer_flashvsr_v1.1_tiny.py:87
↓ 1 callersFunctionprepare_input_tensor
(path: str, scale: int = 4, dtype=torch.bfloat16, device='cuda')
examples/WanVSR/infer_flashvsr_full.py:66
↓ 1 callersFunctionprepare_input_tensor
(path: str, scale: float = 4, dtype=torch.bfloat16, device='cuda')
examples/WanVSR/infer_flashvsr_tiny_long_video.py:87
↓ 1 callersFunctionprepare_input_tensor
(path: str, scale: int = 4, dtype=torch.bfloat16, device='cuda')
examples/WanVSR/infer_flashvsr_v1.1_full.py:66
↓ 1 callersFunctionprepare_input_tensor
(path: str, scale: float = 4, dtype=torch.bfloat16, device='cuda')
examples/WanVSR/infer_flashvsr_tiny.py:87
↓ 1 callersFunctionprepare_inputs_for_generation
(input_ids, image_inputs, past=None, **kwargs)
diffsynth/extensions/ImageQualityMetric/open_clip/coca_model.py:439
↓ 1 callersMethodprepare_inputs_for_generation
( self, input_ids: torch.LongTensor, past_key_values: Optional[torch.Tenso
diffsynth/models/kolors_text_encoder.py:1152
↓ 1 callersMethodprepare_ipadapter
(self, ipadapter_images, ipadapter_scale)
diffsynth/pipelines/flux_image.py:317
↓ 1 callersMethodprepare_ipadapter_inputs
(self, images, height=384, width=384)
diffsynth/pipelines/flux_image.py:263
↓ 1 callersMethodprepare_latents
(self, input_image, height, width, seed, tiled, tile_size, tile_stride)
diffsynth/pipelines/flux_image.py:304
↓ 1 callersMethodprepare_motion_bucket_id
(self, motion_bucket_id)
diffsynth/pipelines/wan_video.py:277
↓ 1 callersMethodprepare_prompts
(self, prompt, local_prompts, masks, mask_scales, t5_sequence_length, negative_prompt, cfg_scale)
diffsynth/pipelines/flux_image.py:361
↓ 1 callersMethodprepare_step1x_kwargs
(self, prompt, negative_prompt, image)
diffsynth/pipelines/flux_image.py:408
↓ 1 callersMethodprepare_text_emb
(self, text_emb, text_emb_t5, text_emb_mask, text_emb_mask_t5)
diffsynth/models/hunyuan_dit.py:262
↓ 1 callersMethodprepare_unified_sequence_parallel
(self)
diffsynth/pipelines/wan_video.py:273
↓ 1 callersMethodprepare_vace_kwargs
( self, latents, vace_video=None, vace_mask=None, vace_reference_image=None, vace_scal
diffsynth/pipelines/wan_video.py:282
↓ 1 callersMethodprepare_vae_images_inputs
(self, semantic_images, i2v_resolution="720p")
diffsynth/pipelines/hunyuan_video.py:78
↓ 1 callersMethodpreprocess_masks
(self, masks, height, width, dim)
diffsynth/pipelines/flux_image.py:281
↓ 1 callersMethodprocess_attention
(self, hidden_states, image_rotary_emb, attn_mask=None, ipadapter_kwargs_list=None)
diffsynth/models/flux_dit.py:229
↓ 1 callersMethodprocess_image
(self, image)
diffsynth/prompters/omnigen_prompter.py:72
↓ 1 callersMethodprocess_image
(self, image)
diffsynth/extensions/ESRGAN/__init__.py:94
↓ 1 callersMethodprocess_image
(self, image)
diffsynth/extensions/RIFE/__init__.py:130
↓ 1 callersMethodprocess_images
(self, images)
diffsynth/extensions/ESRGAN/__init__.py:98
↓ 1 callersMethodprocess_qkv
(self, hidden_states, to_qkv, norm_q, norm_k)
diffsynth/models/sd3_dit.py:173
↓ 1 callersMethodprocess_qkv
(self, q, k, v, image_rotary_emb, text_seq_length)
diffsynth/models/cog_dit.py:72
↓ 1 callersMethodpropagation
(self, source_guide, target_guide, source_style, target_style, nnf, err)
diffsynth/extensions/FastBlend/patch_match.py:165
↓ 1 callersMethodprune_heads
(self, heads)
diffsynth/extensions/ImageQualityMetric/BLIP/med.py:241
↓ 1 callersFunctionpush_pretrained_to_hf_hub
( model_name, pretrained: str, repo_id: str, image_mean: Optional[Tuple[float, ...]] = None,
diffsynth/extensions/ImageQualityMetric/open_clip/push_to_hf_hub.py:124
↓ 1 callersFunctionpush_to_hf_hub
( model, tokenizer, model_config: Optional[dict], repo_id: str, commit_message: str = 'Add
diffsynth/extensions/ImageQualityMetric/open_clip/push_to_hf_hub.py:65
↓ 1 callersMethodquantize
(self, weight_bit_width: int)
diffsynth/models/kolors_text_encoder.py:1103
↓ 1 callersMethodquantize
(self, bits: int, empty_init=False, device=None, **kwargs)
diffsynth/models/kolors_text_encoder.py:1449
↓ 1 callersMethodquickGELU
(self, x)
diffsynth/models/sd_text_encoder.py:16
↓ 1 callersMethodrandom_search
(self, source_guide, target_guide, source_style, target_style, nnf, err)
diffsynth/extensions/FastBlend/patch_match.py:171
↓ 1 callersMethodrandom_step
(self, nnf, r)
diffsynth/extensions/FastBlend/patch_match.py:94
↓ 1 callersMethodrelease_vram
(self)
diffsynth/processors/FastBlend.py:124
↓ 1 callersMethodreparameterize
(self, mu, log_var)
diffsynth/models/wan_video_vae.py:604
↓ 1 callersMethodrescale_zero_terminal_snr
(self, alphas_cumprod)
diffsynth/schedulers/ddim.py:22
↓ 1 callersMethodreshape_for_broadcast
(self, freqs_cis, x)
diffsynth/models/hunyuan_dit.py:16
↓ 1 callersMethodresize_with_antialiasing
(self, input, size, interpolation="bicubic", align_corners=True)
diffsynth/pipelines/svd_video.py:196
↓ 1 callersMethodreverse
(windows: torch.Tensor, win: Tuple[int, int, int], orig: Tuple[int, int, int])
diffsynth/models/wan_video_dit.py:105
↓ 1 callersFunctionrms_norm
(input, normalized_shape, eps=1e-6)
diffsynth/models/stepvideo_vae.py:871
↓ 1 callersMethodrope
(self, pos: torch.Tensor, dim: int, theta: int)
diffsynth/models/flux_dit.py:23
↓ 1 callersFunctionrotate_half
(x)
diffsynth/extensions/ImageQualityMetric/trainer/models/cross_modeling.py:55
↓ 1 callersMethodrotate_half
(x)
diffsynth/models/stepvideo_dit.py:385
↓ 1 callersMethodsample
(self, generator=None)
diffsynth/models/stepvideo_vae.py:895
← previousnext →701–800 of 2,110, ranked by callers