MCPcopy Create free account

hub / github.com/OpenImagingLab/FlashVSR / functions

Functions2,110 in github.com/OpenImagingLab/FlashVSR

↓ 291 callersMethodto
(self,device)
diffsynth/controlnets/processors.py:42
↓ 230 callersMethodto
(self, device)
diffsynth/models/model_manager.py:451
↓ 94 callersMethodfetch_model
(self, model_name, file_path=None, require_model_path=False)
diffsynth/models/model_manager.py:429
↓ 72 callersMethodload_models_to_device
(self, loadmodel_names=[])
diffsynth/pipelines/base.py:95
↓ 55 callersFunction_pcfg
(url='', hf_hub='', mean=None, std=None)
diffsynth/extensions/ImageQualityMetric/open_clip/pretrained.py:21
↓ 48 callersMethodcpu
(self)
diffsynth/controlnets/controlnet_unit.py:27
↓ 35 callersMethodnorm
(self, x)
diffsynth/models/wan_video_dit.py:282
↓ 29 callersMethodgenerate_noise
(self, shape, seed=None, device="cpu", dtype=torch.float16)
diffsynth/pipelines/base.py:124
↓ 28 callersMethodload
(self, model, state_dict_lora, lora_prefix, alpha=1.0, model_resource=None)
diffsynth/models/lora.py:76
↓ 26 callersFunction_n2p
(w, t=True)
diffsynth/extensions/ImageQualityMetric/BLIP/vit.py:203
↓ 26 callersMethodinterpolate
(self, images, scale=1.0, batch_size=4, num_iter=1, progress_bar=lambda x:x)
diffsynth/extensions/RIFE/__init__.py:171
↓ 25 callersMethodpreprocess_image
(self, image)
diffsynth/pipelines/base.py:30
↓ 23 callersMethoddenoising_model
(self)
diffsynth/pipelines/sd_image.py:31
↓ 22 callersMethodfrom_pretrained
(cls, model_name)
diffsynth/prompters/omnigen_prompter.py:61
↓ 19 callersMethod__init__
(self, in_channels )
diffsynth/models/stepvideo_vae.py:467
↓ 19 callersFunctionenable_vram_management
(model: torch.nn.Module, module_map: dict, module_config: dict, max_num_param=None, overflow_module_config: di
diffsynth/vram_management/layers.py:92
↓ 18 callersMethodfrom_pretrained
(file_path, torch_dtype=torch.bfloat16)
diffsynth/models/cog_dit.py:333
↓ 17 callersMethodset_timesteps
(self, num_inference_steps, denoising_strength=1.0, **kwargs)
diffsynth/schedulers/ddim.py:41
↓ 17 callersMethodstep
(self, model_output, timestep, sample, to_final=False)
diffsynth/schedulers/ddim.py:67
↓ 16 callersFunctioncreate_custom_forward
(module)
diffsynth/distributed/xdit_context_parallel.py:70
↓ 16 callersFunctionhash_state_dict_keys
(state_dict, with_shape=True)
diffsynth/models/utils.py:179
↓ 15 callersMethod__init__
(self, dim_in: int, dim_out: int, approximate: str = "none", bias: bool = True)
diffsynth/models/stepvideo_dit.py:571
↓ 15 callersMethodupdate
(self, hidden_states)
diffsynth/pipelines/wan_video.py:514
↓ 14 callersMethodadd_noise
(self, original_samples, noise, timestep)
diffsynth/schedulers/ddim.py:87
↓ 13 callersMethod__init__
(self, config, layer_num)
diffsynth/extensions/ImageQualityMetric/BLIP/med.py:313
↓ 13 callersMethod__init__
(self, patch_size=(1, 2, 2), in_channels=16, embed_dim=3072)
diffsynth/models/hunyuan_video_dit.py:197
↓ 13 callersMethod__init__
(self, config: ChatGLMConfig, device=None)
diffsynth/models/kolors_text_encoder.py:725
↓ 13 callersFunctioncast_to
(weight, dtype, device)
diffsynth/vram_management/layers.py:5
↓ 13 callersMethodcheck_resize_height_width
(self, height, width)
diffsynth/pipelines/base.py:20
↓ 12 callersFunctionload_state_dict
(file_path, torch_dtype=None)
diffsynth/models/utils.py:65
↓ 11 callersMethod__init__
(self, embed_dim=512, image_size=224, patch_size=16,
diffsynth/models/wan_video_image_encoder.py:483
↓ 11 callersMethodget_command
(self, token)
diffsynth/prompters/kolors_prompter.py:114
↓ 10 callersFunctionconv
(in_planes, out_planes, kernel_size=3, stride=1, padding=1, dilation=1)
diffsynth/extensions/RIFE/__init__.py:26
↓ 10 callersFunctionget_model_path
(model_name)
diffsynth/extensions/ImageQualityMetric/config.py:8
↓ 10 callersMethodpatchify
(self, hidden_states)
diffsynth/models/flux_dit.py:297
↓ 10 callersMethodprocess_prompt
(self, prompt, positive=True)
diffsynth/prompters/base_prompter.py:57
↓ 9 callersMethod__init__
( self, dim, num_heads=8, qkv_bias=True, scaled_co
diffsynth/extensions/ImageQualityMetric/open_clip/transformer.py:88
↓ 9 callersMethod__init__
(self, embed_dim=1536, num_layers=24, use_rms_norm=False, num_dual_blocks=0, pos_embed_max_size=192)
diffsynth/models/sd3_dit.py:327
↓ 9 callersMethod__init__
(self, dim_in, dim_out)
diffsynth/models/sd_unet.py:22
↓ 9 callersMethod__init__
(self, disable_guidance_embedder=False, input_dim=64, num_blocks=19)
diffsynth/models/flux_dit.py:279
↓ 9 callersMethodencode
(self, s: str, bos: bool = False, eos: bool = False)
diffsynth/prompters/kolors_prompter.py:50
↓ 9 callersMethodunpatchify
(self, hidden_states, height, width)
diffsynth/models/cog_dit.py:230
↓ 8 callersMethod__init__
( self, in_channels, hidden_channels=None, out_features=None, act_laye
diffsynth/models/step1x_connector.py:25
↓ 8 callersMethod__init__
(self, input_ids, cu_input_ids, attention_mask, cu_seqlens, max_seq_len)
diffsynth/models/stepvideo_text_encoder.py:185
↓ 8 callersMethod__init__
(self, dim, channel_first=True, images=True, bias=False)
diffsynth/models/wan_video_vae.py:58
↓ 8 callersMethod__init__
(self, in_dim, out_dim, has_pos_emb=False)
diffsynth/models/wan_video_dit.py:479
↓ 8 callersMethod_to_tuple
(self, x)
diffsynth/pipelines/hunyuan_image.py:20
↓ 8 callersFunctionconv
(n_in, n_out, **kwargs)
examples/WanVSR/utils/TCDecoder.py:34
↓ 8 callersMethodfrom_model_manager
(model_manager: ModelManager)
diffsynth/prompters/omost.py:259
↓ 8 callersMethodload_models
(self, file_path_list, model_names=None, device=None, torch_dtype=None)
diffsynth/models/model_manager.py:424
↓ 8 callersMethodload_prompt_refiners
(self, model_manager: ModelManager, refiner_classes=[])
diffsynth/prompters/base_prompter.py:45
↓ 8 callersMethodpreprocess_images
(self, images)
diffsynth/pipelines/base.py:35
↓ 8 callersMethodprocess_image
(self, image)
diffsynth/processors/RIFE.py:24
↓ 8 callersMethodsave
(self, path)
diffsynth/extensions/ImageQualityMetric/trainer/models/clip_model.py:144
↓ 8 callersMethodto
(self, device)
diffsynth/models/stepvideo_text_encoder.py:191
↓ 7 callersMethod__init__
(self, in_channels, num_heads, head_dim, n
diffsynth/models/hunyuan_video_vae_decoder.py:120
↓ 7 callersMethodenable_cpu_offload
(self)
diffsynth/pipelines/base.py:91
↓ 7 callersMethodencode_prompt
(self, prompt, clip_skip=1, positive=True)
diffsynth/pipelines/sd_image.py:81
↓ 7 callersFunctioninit_weights_on_device
(device = torch.device("meta"), include_buffers :bool = False)
diffsynth/models/utils.py:7
↓ 7 callersMethodtiled_forward
(self, hidden_states, timestep, prompt_emb, pooled_prompt_emb, tile_size=128, tile_stride=64)
diffsynth/models/sd3_dit.py:339
↓ 7 callersMethodtranspose_for_scores
(self, x)
diffsynth/extensions/ImageQualityMetric/BLIP/med.py:130
↓ 6 callersMethod__init__
( self, in_channels: int, out_channels: int, kernel_size: int = 3, str
diffsynth/models/cog_vae.py:58
↓ 6 callersMethod__init__
(self, add_positional_conv=None)
diffsynth/models/svd_unet.py:239
↓ 6 callersMethod__init__
( self, patch_size=(2, 2), in_chans=4, embed_dim=1408, bias=True,
diffsynth/models/hunyuan_dit.py:177
↓ 6 callersMethod__init__
(self, n_f, stride)
examples/WanVSR/utils/TCDecoder.py:55
↓ 6 callersFunction_to_tuple
(x, dim=2)
diffsynth/models/hunyuan_video_dit.py:11
↓ 6 callersFunctionattention
(q, k, v)
diffsynth/models/hunyuan_video_dit.py:387
↓ 6 callersMethodbuild_remapping_table
(self, frames_guide, frames_style, patch_match_engine, batch_size, desc="")
diffsynth/extensions/FastBlend/runners/fast.py:34
↓ 6 callersFunctioncheck_is_instance
(model, module_class)
diffsynth/models/wan_video_vae.py:11
↓ 6 callersMethodclear_cache
(self)
diffsynth/models/cog_vae.py:175
↓ 6 callersMethoddecode_video
(self, latents, tiled=False, tile_size=64, tile_stride=32)
diffsynth/pipelines/sd_video.py:125
↓ 6 callersMethodenable_vram_management
(self, num_persistent_param_in_dit=None)
diffsynth/pipelines/wan_video.py:44
↓ 6 callersMethodestimate_nnf
(self, source_guide, target_guide, source_style)
diffsynth/extensions/FastBlend/patch_match.py:282
↓ 6 callersMethodinit_cross_kv
( self, context_tensor: Optional[torch.Tensor] = None, )
diffsynth/pipelines/flashvsr_tiny.py:229
↓ 6 callersMethodload
(self, file_path="", state_dict={}, device="cuda", torch_dtype=torch.float16, **kwargs)
diffsynth/models/model_manager.py:143
↓ 6 callersMethodmatch
(self, file_path="", state_dict={})
diffsynth/models/model_manager.py:140
↓ 6 callersMethodprocess_window_sum
(self, frames_guide, blending_table, patch_match_engine, window_size, batch_size, desc="")
diffsynth/extensions/FastBlend/runners/fast.py:77
↓ 6 callersMethodremapping_table_to_blending_table
(self, table)
diffsynth/extensions/FastBlend/runners/fast.py:56
↓ 6 callersMethodrun
(self, frames_guide, frames_style, batch_size, window_size, ebsynth_config)
diffsynth/extensions/FastBlend/__init__.py:26
↓ 6 callersFunctionsearch_for_images
(folder)
diffsynth/extensions/FastBlend/data.py:65
↓ 6 callersFunctionsinusoidal_embedding_1d
(dim, position)
diffsynth/models/wan_video_dit.py:243
↓ 6 callersMethodstate_dict_converter
()
diffsynth/models/sd3_dit.py:381
↓ 6 callersMethodstream_forward
(self, video_clip)
examples/WanVSR/utils/utils.py:132
↓ 6 callersMethodvae_output_to_image
(self, vae_output)
diffsynth/pipelines/base.py:39
↓ 5 callersMethod__init__
(self, fn)
diffsynth/extensions/ImageQualityMetric/trainer/models/cross_modeling.py:31
↓ 5 callersMethod__init__
(self, dim, eps=1e-6)
diffsynth/models/wan_video_text_encoder.py:24
↓ 5 callersFunction_ntuple
(n)
diffsynth/extensions/ImageQualityMetric/open_clip/utils.py:48
↓ 5 callersFunctionclosest_name
(input_str, options)
diffsynth/prompters/omost.py:98
↓ 5 callersMethodcontrol_noise_via_local_prompts
(self, prompt_emb_global, prompt_emb_locals, masks, mask_scales, inference_callback, special_kwargs=None, spec
diffsynth/pipelines/base.py:66
↓ 5 callersMethoddecode
(self, t: List[int])
diffsynth/prompters/kolors_prompter.py:59
↓ 5 callersMethoddecode
(self, z, scale)
diffsynth/models/wan_video_vae.py:553
↓ 5 callersMethodencode
(self, x, scale)
diffsynth/models/wan_video_vae.py:526
↓ 5 callersMethodencode_image
(self, image, tiled=False, tile_size=64, tile_stride=32)
diffsynth/pipelines/flux_image.py:194
↓ 5 callersMethodencode_video
(self, input_video, tiled=True, tile_size=(34, 34), tile_stride=(18, 16))
diffsynth/pipelines/wan_video.py:263
↓ 5 callersMethodgenerate
( self, image, text=None, seq_len=30, max_seq_len=77, temperat
diffsynth/extensions/ImageQualityMetric/open_clip/coca_model.py:167
↓ 5 callersMethodget_text_features
(self, *args, **kwargs)
diffsynth/extensions/ImageQualityMetric/trainer/models/clip_model.py:110
↓ 5 callersFunctionmodulate
(x, shift=None, scale=None, tr_shift=None, tr_scale=None, tr_token=None)
diffsynth/models/hunyuan_video_dit.py:285
↓ 5 callersMethodupdate
(self, source_guide, target_guide, source_style, target_style, nnf, err, upd_nnf)
diffsynth/extensions/FastBlend/patch_match.py:158
↓ 4 callersMethod__init__
(self, hidden_size, patch_size, out_channels)
diffsynth/models/omnigen.py:239
↓ 4 callersMethod__init__
(self)
diffsynth/models/lora.py:16
next →1–100 of 2,110, ranked by callers