MCPcopy Create free account

hub / github.com/amap-cvlab/ABot-PhysWorld / functions

Functions2,967 in github.com/amap-cvlab/ABot-PhysWorld

⨍ Functions 2,967◇ Types & classes 770↳ Endpoints 1
↓ 543 callersMethodto
(self, *args, **kwargs)
inference/diffsynth/utils/__init__.py:33
↓ 330 callersMethodto
(self, device)
inference/diffsynth/models/model_manager.py:464
↓ 151 callersMethodload_models_to_device
(self, loadmodel_names=[])
inference/diffsynth/pipelines/base.py:95
↓ 134 callersMethodfetch_model
(self, model_name, file_path=None, require_model_path=False, index=None)
inference/diffsynth/models/model_manager.py:429
↓ 87 callersMethodcpu
(self)
inference/diffsynth/controlnets/controlnet_unit.py:27
↓ 81 callersMethodpreprocess_image
(self, image)
inference/diffsynth/pipelines/base.py:30
↓ 75 callersFunctionprint0
(*args, **kwargs)
EZS-Bench/pbench/distributed.py:19
↓ 72 callersMethodencode
(self, s: str, bos: bool = False, eos: bool = False)
inference/diffsynth/prompters/kolors_prompter.py:50
↓ 55 callersFunction_pcfg
(url='', hf_hub='', mean=None, std=None)
inference/diffsynth/extensions/ImageQualityMetric/open_clip/pretrained.py:21
↓ 55 callersFunctionenable_vram_management
(model: torch.nn.Module, module_map: dict, module_config: dict, max_num_param=None, overflow_module_config: di
inference/diffsynth/vram_management/layers.py:210
↓ 51 callersMethodpreprocess_video
(self, video, torch_dtype=None, device=None, pattern="B C T H W", min_value=-1, max_value=1)
inference/diffsynth/utils/__init__.py:69
↓ 46 callersMethodinterpolate
(self, images, scale=1.0, batch_size=4, num_iter=1, progress_bar=lambda x:x)
inference/diffsynth/extensions/RIFE/__init__.py:171
↓ 45 callersMethodload
(self, model, state_dict_lora, lora_prefix, alpha=1.0, model_resource=None)
inference/diffsynth/models/lora.py:76
↓ 43 callersMethodnorm
(self, x)
inference/diffsynth/models/wan_video_dit.py:106
↓ 34 callersFunctionload_state_dict
(file_path, torch_dtype=None, device="cpu")
inference/diffsynth/models/utils.py:65
↓ 31 callersFunctionhash_state_dict_keys
(state_dict, with_shape=True)
inference/diffsynth/models/utils.py:206
↓ 31 callersMethodupdate
(self, hidden_states)
inference/diffsynth/pipelines/wan_video.py:531
↓ 28 callersMethodadd_noise
(self, original_samples, noise, timestep)
inference/diffsynth/schedulers/ddim.py:87
↓ 28 callersMethodgenerate_noise
(self, shape, seed=None, device="cpu", dtype=torch.float16)
inference/diffsynth/pipelines/base.py:124
↓ 28 callersMethodrun
(self, model_fn, sliding_window_size, sliding_window_stride, computation_device, computation_dtype, model_kwar
inference/diffsynth/pipelines/wan_video_new.py:1488
↓ 27 callersMethodstep
(self, scheduler, latents, progress_id, noise_pred, input_latents=None, inpaint_mask=None, **kwargs)
inference/diffsynth/utils/__init__.py:148
↓ 26 callersFunction_n2p
(w, t=True)
inference/diffsynth/extensions/ImageQualityMetric/BLIP/vit.py:203
↓ 26 callersFunctioncreate_custom_forward
(module)
inference/diffsynth/pipelines/wan_video_new.py:1716
↓ 26 callersMethodfrom_pretrained
(file_path, torch_dtype=torch.bfloat16)
inference/diffsynth/models/cog_dit.py:333
↓ 25 callersMethod__init__
(self, num_inference_steps, rel_l1_thresh, model_id)
inference/diffsynth/pipelines/wan_video_new.py:1414
↓ 25 callersMethod__init__
(self, num_inference_steps, rel_l1_thresh, model_id)
inference/diffsynth/pipelines/wan_video_new_vace.py:1429
↓ 25 callersMethod__init__
(self, num_inference_steps, rel_l1_thresh, model_id)
inference/diffsynth/pipelines/wan_video_new_bak.py:1255
↓ 25 callersFunctionget_rank
()
EZS-Bench/pbench/distributed.py:15
↓ 24 callersMethodencode_image
(self, image, tiled=False, tile_size=64, tile_stride=32)
inference/diffsynth/pipelines/sd_image.py:70
↓ 24 callersMethodsave
(self, path)
inference/diffsynth/extensions/ImageQualityMetric/trainer/models/clip_model.py:144
↓ 24 callersMethodset_timesteps
(self, num_inference_steps, denoising_strength=1.0, **kwargs)
inference/diffsynth/schedulers/ddim.py:41
↓ 22 callersMethodpatchify
(self, hidden_states)
inference/diffsynth/models/flux_dit.py:297
↓ 21 callersMethoddownload_if_necessary
(self, use_usp=False)
inference/diffsynth/utils/__init__.py:169
↓ 21 callersMethodfrom_pretrained
(cls, model_name)
inference/diffsynth/prompters/omnigen_prompter.py:61
↓ 20 callersMethod__init__
(self, num_inference_steps, rel_l1_thresh)
inference/diffsynth/pipelines/flux_image_new.py:1066
↓ 19 callersMethod__init__
(self, in_channels )
inference/diffsynth/models/stepvideo_vae.py:467
↓ 18 callersMethodencode_prompt
(self, prompt, clip_skip=1, positive=True)
inference/diffsynth/pipelines/sd_image.py:81
↓ 17 callersMethod__init__
( self, in_channels: int, out_channels: int, factor_t, factor_s=1,
inference/diffsynth/models/wan_video_vae.py:412
↓ 17 callersMethod__init__
(self, kernel, pad, upsample_factor=1)
inference/diffsynth/models/wan_video_animate_adapter.py:370
↓ 16 callersMethodcheck_resize_height_width
(self, height, width)
inference/diffsynth/pipelines/base.py:20
↓ 16 callersFunctionsinusoidal_embedding_1d
(dim, position)
inference/diffsynth/models/wan_video_dit.py:68
↓ 15 callersMethod__init__
(self, dim_in: int, dim_out: int, approximate: str = "none", bias: bool = True)
inference/diffsynth/models/stepvideo_dit.py:571
↓ 15 callersMethodafter_transformer_block
(self, block_idx, hidden_states, audio_emb_global, audio_emb, original_seq_len, use_unified_sequence_parallel=
inference/diffsynth/models/wan_video_dit_s2v.py:460
↓ 15 callersFunctioncast_to
(weight, dtype, device)
inference/diffsynth/vram_management/layers.py:5
↓ 13 callersMethod__init__
(self, config, layer_num)
inference/diffsynth/extensions/ImageQualityMetric/BLIP/med.py:313
↓ 13 callersMethod__init__
(self, patch_size=(1, 2, 2), in_channels=16, embed_dim=3072)
inference/diffsynth/models/hunyuan_video_dit.py:197
↓ 13 callersMethod__init__
(self, config: ChatGLMConfig, device=None)
inference/diffsynth/models/kolors_text_encoder.py:725
↓ 13 callersMethodunpatchify
(self, hidden_states, height, width)
inference/diffsynth/models/cog_dit.py:230
↓ 12 callersMethodrope_params
Args: index: [0, 1, 2, 3] 1D Tensor representing the position index of the token
inference/diffsynth/models/qwen_image_dit.py:81
↓ 12 callersMethodupdate
(self, hidden_states)
inference/diffsynth/pipelines/wan_video_new.py:1460
↓ 12 callersMethodupdate
(self, hidden_states)
inference/diffsynth/pipelines/wan_video_new_vace.py:1475
↓ 12 callersMethodupdate
(self, hidden_states)
inference/diffsynth/pipelines/wan_video_new_bak.py:1301
↓ 11 callersMethod__init__
(self, embed_dim=512, image_size=224, patch_size=16,
inference/diffsynth/models/wan_video_image_encoder.py:483
↓ 11 callersMethod__init__
( self, dim: int, num_heads: int, enable_flashattn3: bool = False, ena
inference/diffsynth/models/longcat_video_dit.py:138
↓ 11 callersFunctionconv
(in_planes, out_planes, kernel_size=3, stride=1, padding=1, dilation=1)
inference/diffsynth/extensions/RIFE/__init__.py:26
↓ 11 callersMethodget_command
(self, token)
inference/diffsynth/prompters/kolors_prompter.py:114
↓ 11 callersFunctionimage_transform
( image_size: int, is_train: bool, mean: Optional[Tuple[float, ...]] = None, s
inference/diffsynth/extensions/ImageQualityMetric/open_clip/transform.py:77
↓ 10 callersMethod__init__
(self, device="cuda", torch_dtype=torch.bfloat16)
inference/diffsynth/pipelines/qwen_image.py:50
↓ 10 callersMethod__init__
(self, embed_dim=1536, num_layers=24, use_rms_norm=False, num_dual_blocks=0, pos_embed_max_size=192)
inference/diffsynth/models/sd3_dit.py:343
↓ 10 callersFunctionget_model_path
(model_name)
inference/diffsynth/extensions/ImageQualityMetric/config.py:8
↓ 10 callersMethodprocess_prompt
(self, prompt, positive=True)
inference/diffsynth/prompters/base_prompter.py:57
↓ 10 callersMethodstate_dict_converter
()
inference/diffsynth/models/sd3_dit.py:397
↓ 10 callersMethodtraining_target
(self, sample, noise, timestep)
inference/diffsynth/schedulers/ddim.py:94
↓ 9 callersMethod__init__
( self, dim, num_heads=8, qkv_bias=True, scaled_co
inference/diffsynth/extensions/ImageQualityMetric/open_clip/transformer.py:88
↓ 9 callersMethod__init__
(self, dim_in, dim_out)
inference/diffsynth/models/sd_unet.py:22
↓ 9 callersMethod__init__
( self, base_dim: int = 96, z_dim: int = 16, dim_mult: Tuple[int] = [1, 2, 4,
inference/diffsynth/models/qwen_image_vae.py:641
↓ 9 callersMethod__init__
(self, disable_guidance_embedder=False, input_dim=64, num_blocks=19)
inference/diffsynth/models/flux_dit.py:279
↓ 9 callersFunctiondistribute_list_to_rank
(data_list)
EZS-Bench/pbench/distributed.py:123
↓ 9 callersFunctiongather_list_of_dict
(results)
EZS-Bench/pbench/distributed.py:117
↓ 9 callersFunctionget_world_size
()
EZS-Bench/pbench/distributed.py:11
↓ 9 callersMethodpreprocess_images
(self, images)
inference/diffsynth/pipelines/base.py:35
↓ 8 callersMethod__init__
( self, in_channels, hidden_channels=None, out_features=None, act_laye
inference/diffsynth/models/step1x_connector.py:25
↓ 8 callersMethod__init__
(self, input_ids, cu_input_ids, attention_mask, cu_seqlens, max_seq_len)
inference/diffsynth/models/stepvideo_text_encoder.py:191
↓ 8 callersMethod__init__
(self, in_dim, out_dim, has_pos_emb=False)
inference/diffsynth/models/wan_video_dit.py:234
↓ 8 callersMethod_to_tuple
(self, x)
inference/diffsynth/pipelines/hunyuan_image.py:20
↓ 8 callersFunctionbarrier
()
EZS-Bench/pbench/distributed.py:105
↓ 8 callersMethodencode
(self, x, scale)
inference/diffsynth/models/wan_video_vae.py:997
↓ 8 callersFunctioninit_weights_on_device
(device = torch.device("meta"), include_buffers :bool = False)
inference/diffsynth/models/utils.py:7
↓ 8 callersMethodload_prompt_refiners
(self, model_manager: ModelManager, refiner_classes=[])
inference/diffsynth/prompters/base_prompter.py:45
↓ 8 callersMethodprepare_image_ids
(self, latents)
inference/diffsynth/models/flux_dit.py:307
↓ 8 callersMethodprocess_image
(self, image)
inference/diffsynth/processors/RIFE.py:24
↓ 8 callersMethodtiled_forward
(self, hidden_states, timestep, prompt_emb, pooled_prompt_emb, tile_size=128, tile_stride=64)
inference/diffsynth/models/sd3_dit.py:355
↓ 8 callersMethodto
(self, device)
inference/diffsynth/models/stepvideo_text_encoder.py:197
↓ 8 callersMethodupdate
(self, hidden_states)
inference/diffsynth/pipelines/flux_image_new.py:1103
↓ 8 callersMethodvae_output_to_image
(self, vae_output)
inference/diffsynth/pipelines/base.py:39
↓ 7 callersMethod__init__
(self, in_channels, num_heads, head_dim, n
inference/diffsynth/models/hunyuan_video_vae_decoder.py:120
↓ 7 callersFunctioncheck_is_instance
(model, module_class)
inference/diffsynth/models/wan_video_vae.py:11
↓ 7 callersMethoddecode
(self, z, scale)
inference/diffsynth/models/wan_video_vae.py:1024
↓ 7 callersMethoddenoising_model
(self)
inference/diffsynth/pipelines/sd_image.py:31
↓ 7 callersFunctionflash_attention
(q: torch.Tensor, k: torch.Tensor, v: torch.Tensor, num_heads: int, compatibility_mode=False)
inference/diffsynth/models/wan_video_dit.py:28
↓ 7 callersMethodto
(self, *args, **kwargs)
inference/diffsynth/trainers/utils.py:378
↓ 7 callersMethodtranspose_for_scores
(self, x)
inference/diffsynth/extensions/ImageQualityMetric/BLIP/med.py:130
↓ 6 callersMethod__init__
( self, in_channels: int, out_channels: int, kernel_size: int = 3, str
inference/diffsynth/models/cog_vae.py:58
↓ 6 callersMethod__init__
(self, config)
inference/diffsynth/models/nexus_gen_projector.py:182
↓ 6 callersMethod__init__
( self, dim: int, in_dim: int, ffn_dim: int, out_dim: int, tex
inference/diffsynth/models/wan_video_dit_s2v.py:362
↓ 6 callersMethod__init__
(self, add_positional_conv=None)
inference/diffsynth/models/svd_unet.py:246
↓ 6 callersMethod__init__
( self, patch_size=(2, 2), in_chans=4, embed_dim=1408, bias=True,
inference/diffsynth/models/hunyuan_dit.py:177
↓ 6 callersFunction_to_tuple
(x, dim=2)
inference/diffsynth/models/hunyuan_video_dit.py:11
↓ 6 callersFunctionattention
(q, k, v)
inference/diffsynth/models/hunyuan_video_dit.py:387
↓ 6 callersMethodbuild_remapping_table
(self, frames_guide, frames_style, patch_match_engine, batch_size, desc="")
inference/diffsynth/extensions/FastBlend/runners/fast.py:34
next →1–100 of 2,967, ranked by callers