MCPcopy Create free account

hub / github.com/bilibili/Index-anisora / functions

Functions18,851 in github.com/bilibili/Index-anisora

⨍ Functions 18,851◇ Types & classes 4,392↳ Endpoints 34
↓ 7 callersFunctionNormalize
(in_channels, num_groups=32)
anisoraV1_train_gpu/sat/sgm/modules/diffusionmodules/model.py:49
↓ 7 callersMethod__get_session
(self, session_id: str)
reward/character/samurai/sam2/demo/backend/server/inference/predictor.py:391
↓ 7 callersMethod__init__
(self, dim=128, z_dim=4, dim_mult=[1, 2, 4, 4],
anisoraV3.2/wan/modules/vae2_1.py:485
↓ 7 callersMethod__init__
(self)
anisora_rl/SwissArmyTransformer-main/sat/model/official/glm130B_model.py:85
↓ 7 callersMethod__init__
(self)
anisora_rl/sat/sgm/modules/autoencoding/losses/video_loss.py:81
↓ 7 callersMethod__init__
(self, **kwargs)
anisora_rl/sat/vae_modules/autoencoder.py:505
↓ 7 callersMethod__init__
(self, dim, out_dim, patch_size, eps=1e-6)
anisora_anymask/wan/modules/model_infer.py:319
↓ 7 callersMethod__init__
(self, dim, out_dim, patch_size, eps=1e-6)
anisora_anymask/wan/modules/model.py:400
↓ 7 callersMethod__init__
(self, dim, out_dim, patch_size, eps=1e-6)
anisoraV3/wan/modules/model_infer.py:319
↓ 7 callersMethod__init__
(self, dim, out_dim, patch_size, eps=1e-6)
anisoraV3/wan/modules/model.py:400
↓ 7 callersMethod__init__
(self)
anisoraV1_train_npu/swissarmytransformer-npu_t_sp/sat/model/official/glm130B_model.py:85
↓ 7 callersMethod__init__
(self)
anisoraV1_train_npu/sgm/modules/autoencoding/losses/video_loss.py:88
↓ 7 callersMethod__init__
(self)
anisoraV1_train_npu/sat/sgm/modules/autoencoding/losses/video_loss.py:81
↓ 7 callersMethod__init__
(self, dim, out_dim, patch_size, eps=1e-6)
anisoraV2_gpu/wan/modules/model_infer.py:319
↓ 7 callersMethod__init__
(self, dim, out_dim, patch_size, eps=1e-6)
anisoraV2_gpu/wan/modules/model.py:400
↓ 7 callersMethod__init__
(self)
anisoraV1_infer/fastercache/models/cogvideox/sgm/modules/autoencoding/losses/video_loss.py:81
↓ 7 callersMethod__init__
(self)
anisoraV1_infer/sat/model/official/glm130B_model.py:85
↓ 7 callersMethod__init__
(self)
anisoraV1_infer/sat/sat/model/official/glm130B_model.py:85
↓ 7 callersMethod__init__
(self)
anisoraV1_train_gpu/swissarmytransformer-npu_t_sp/sat/model/official/glm130B_model.py:85
↓ 7 callersMethod__init__
(self)
anisoraV1_train_gpu/sgm/modules/autoencoding/losses/video_loss.py:88
↓ 7 callersMethod__init__
(self)
anisoraV1_train_gpu/sat/sgm/modules/autoencoding/losses/video_loss.py:81
↓ 7 callersMethod__init__
(self, dim, out_dim, patch_size, eps=1e-6)
anisoraV2_npu/fastvideo/bili_space/wan/modules/model_infer.py:344
↓ 7 callersMethod__init__
(self, dim, out_dim, patch_size, eps=1e-6)
anisoraV2_npu/fastvideo/bili_space/wan/modules/model.py:401
↓ 7 callersMethod__len__
(self)
anisoraV1_infer/sat/data_utils/datasets.py:80
↓ 7 callersFunction_broadcast_tensor
(a, broadcast_shape)
anisoraV2_gpu/OSS/utils.py:5
↓ 7 callersMethod_get_decoder_layers
(self)
reward/mantis/mllm_tools/model_utils/otter/models/flamingo/modeling_flamingo.py:407
↓ 7 callersMethod_get_decoder_layers
(self)
reward/mantis/models/openflamingo/flamingo_lm.py:78
↓ 7 callersFunction_is_tensor_video_clip
(clip)
anisoraV1_infer/fastercache/datasets/video_transform.py:19
↓ 7 callersFunctionall_to_all_comm
(input_,scatter_dim=2, gather_dim=1)
anisoraV2_npu/fastvideo/utils/communications.py:514
↓ 7 callersFunctionapply_gate
AI is creating summary for apply_gate Args: x (torch.Tensor): input tensor. gate (torch.Tensor, optional): gate tensor. Defaults
anisoraV2_gpu/fastvideo/models/hunyuan/modules/modulate_layers.py:53
↓ 7 callersFunctionapply_gate
AI is creating summary for apply_gate Args: x (torch.Tensor): input tensor. gate (torch.Tensor, optional): gate tensor. Defaults
anisoraV2_npu/fastvideo/models/hunyuan/modules/modulate_layers.py:53
↓ 7 callersMethodbackward
(ctx, grad_output: torch.Tensor)
anisora_rl/SwissArmyTransformer-main/sat/quantization/kernels.py:61
↓ 7 callersMethodbackward
(ctx, grad_output: torch.Tensor)
anisoraV1_train_npu/swissarmytransformer-npu_t_sp/sat/quantization/kernels.py:61
↓ 7 callersMethodbackward
(ctx, grad_output: torch.Tensor)
anisoraV1_train_gpu/swissarmytransformer-npu_t_sp/sat/quantization/kernels.py:61
↓ 7 callersFunctioncast_tuple
(t, length=1)
anisoraV1_infer/videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v110.py:1259
↓ 7 callersFunctioncast_tuple
(t, length=1)
anisoraV1_infer/fastercache/models/opensora_plan/modules/ops.py:30
↓ 7 callersFunctionconcat_all_gather
Performs all_gather operation on the provided tensors. *** Warning ***: torch.distributed.all_gather has no gradient.
reward/models/blip_retrieval.py:274
↓ 7 callersFunctioncopy_from_param
(src, dst)
anisoraV1_infer/sat/examples/cait/transform_param.py:37
↓ 7 callersFunctioncopy_to_model_parallel_region
(input_)
anisora_rl/SwissArmyTransformer-main/sat/mpu/mappings.py:131
↓ 7 callersFunctioncopy_to_model_parallel_region
(input_)
anisoraV1_train_npu/swissarmytransformer-npu_t_sp/sat/mpu/mappings.py:131
↓ 7 callersFunctioncopy_to_model_parallel_region
(input_)
anisoraV1_infer/sat/sat/mpu/mappings.py:131
↓ 7 callersFunctioncopy_to_model_parallel_region
(input_)
anisoraV1_train_gpu/swissarmytransformer-npu_t_sp/sat/mpu/mappings.py:131
↓ 7 callersFunctioncrop
Args: clip (torch.tensor): Video clip to be cropped. Size is (T, C, H, W)
anisoraV2_gpu/fastvideo/dataset/transform.py:40
↓ 7 callersFunctioncrop
Args: clip (torch.tensor): Video clip to be cropped. Size is (T, C, H, W)
anisoraV2_npu/fastvideo/dataset/transform.py:40
↓ 7 callersMethoddecode
(self, z, scale)
anisoraV3.2/wan/modules/vae2_2.py:812
↓ 7 callersFunctiondefault
(*args)
anisora_rl/sat/sgm/modules/autoencoding/regularizers/lookup_free_quantization.py:33
↓ 7 callersFunctiondefault
(val, d)
anisora_rl/sat/vae_modules_infer/utils.py:225
↓ 7 callersFunctiondefault
(*args)
anisoraV1_train_npu/sgm/modules/autoencoding/regularizers/lookup_free_quantization.py:31
↓ 7 callersFunctiondefault
(*args)
anisoraV1_train_npu/sat/sgm/modules/autoencoding/regularizers/lookup_free_quantization.py:33
↓ 7 callersFunctiondefault
(val, d)
anisoraV1_train_npu/sat/vae_modules/utils.py:225
↓ 7 callersFunctiondefault
(*args)
anisoraV1_infer/fastercache/models/cogvideox/sgm/modules/autoencoding/regularizers/lookup_free_quantization.py:33
↓ 7 callersFunctiondefault
(val, d)
anisoraV1_infer/fastercache/models/cogvideox/vae_modules/utils.py:217
↓ 7 callersFunctiondefault
(*args)
anisoraV1_train_gpu/sgm/modules/autoencoding/regularizers/lookup_free_quantization.py:31
↓ 7 callersFunctiondefault
(*args)
anisoraV1_train_gpu/sat/sgm/modules/autoencoding/regularizers/lookup_free_quantization.py:33
↓ 7 callersFunctiondefault
(val, d)
anisoraV1_train_gpu/sat/vae_modules/utils.py:225
↓ 7 callersMethoddevice
(self)
anisoraV1_infer/videosys/models/autoencoders/autoencoder_kl_open_sora.py:550
↓ 7 callersMethodencode
Args: batch_encoding (dict): Batch encoding from tokenizer. use_attention_mask (bool): Whether to use attention mask.
anisoraV2_gpu/fastvideo/models/hunyuan/text_encoder/__init__.py:271
↓ 7 callersFunctionflash_attn_varlen_func
( q, k, v, cu_seqlens_q, cu_seqlens_k, max_seqlen_q, max_seqlen_k, dropout_p=0.0,
anisoraV2_npu/fastvideo/bili_space/npu_func.py:58
↓ 7 callersMethodforward
Forward pass. Args: x: Input tensor. Shape: [B, C, *] or [B, *, C]. Returns: x: Output tensor. Shape: [B, C'
anisoraV1_infer/fastercache/models/genmo/mochi_preview/vae/models.py:174
↓ 7 callersFunctionget_args
Parse all the args.
anisora_rl/SwissArmyTransformer-main/sat/arguments.py:335
↓ 7 callersFunctionget_args
Parse all the args.
anisoraV1_train_npu/swissarmytransformer-npu_t_sp/sat/arguments.py:337
↓ 7 callersFunctionget_args
Parse all the args.
anisoraV1_train_gpu/swissarmytransformer-npu_t_sp/sat/arguments.py:337
↓ 7 callersFunctionget_context_parallel_group_rank
()
anisoraV1_infer/fastercache/models/cogvideox/vae_modules/utils.py:57
↓ 7 callersFunctionget_context_parallel_rank
()
anisora_rl/sat/sgm/util.py:63
↓ 7 callersFunctionget_context_parallel_rank
()
anisora_rl/sat/vae_modules_infer/utils.py:55
↓ 7 callersFunctionget_context_parallel_rank
()
anisoraV1_train_npu/sgm/util.py:60
↓ 7 callersFunctionget_context_parallel_rank
()
anisoraV1_train_npu/sat/sgm/util.py:62
↓ 7 callersFunctionget_context_parallel_rank
()
anisoraV1_train_npu/sat/vae_modules/utils.py:55
↓ 7 callersFunctionget_context_parallel_rank
()
anisoraV1_infer/fastercache/models/cogvideox/sgm/util.py:62
↓ 7 callersFunctionget_context_parallel_rank
()
anisoraV1_train_gpu/sgm/util.py:60
↓ 7 callersFunctionget_context_parallel_rank
()
anisoraV1_train_gpu/sat/sgm/util.py:62
↓ 7 callersFunctionget_context_parallel_rank
()
anisoraV1_train_gpu/sat/vae_modules/utils.py:55
↓ 7 callersFunctionget_context_parallel_world_size
()
anisora_rl/sat/sgm/util.py:57
↓ 7 callersFunctionget_context_parallel_world_size
()
anisora_rl/sat/vae_modules_infer/utils.py:49
↓ 7 callersFunctionget_context_parallel_world_size
()
anisoraV1_train_npu/sat/vae_modules/utils.py:49
↓ 7 callersFunctionget_context_parallel_world_size
()
anisoraV1_train_gpu/sat/vae_modules/utils.py:49
↓ 7 callersMethodget_latent_size
(self, input_size)
anisoraV1_infer/videosys/models/autoencoders/autoencoder_kl_open_sora.py:424
↓ 7 callersFunctionget_model
Build the model.
anisoraV1_train_npu/swissarmytransformer-npu_t_sp/sat/model/base_model.py:395
↓ 7 callersFunctionget_model
Build the model.
anisoraV1_train_gpu/swissarmytransformer-npu_t_sp/sat/model/base_model.py:395
↓ 7 callersMethodget_model
(self, *, local_rank: int, device_id: Union[int, Literal["cpu"]], world_size: int)
anisoraV1_infer/fastercache/models/genmo/mochi_preview/pipelines.py:123
↓ 7 callersFunctionget_temporal_pad
()
anisoraV1_infer/sat/sequence_parallel/comm.py:375
↓ 7 callersMethodgoToFrame
(index: number)
reward/character/samurai/sam2/demo/frontend/src/common/components/video/VideoWorkerBridge.ts:278
↓ 7 callersFunctioninit_tokenizer
()
reward/models/blip.py:187
↓ 7 callersFunctioninitialize_context_parallel
(context_parallel_size)
anisoraV1_infer/fastercache/models/cogvideox/sgm/util.py:32
↓ 7 callersFunctionload_checkpoint
Load a model checkpoint.
anisoraV1_train_npu/swissarmytransformer-npu_t_sp/sat/training/model_io.py:260
↓ 7 callersFunctionload_checkpoint
(model,url_or_filename)
reward/models/blip.py:216
↓ 7 callersFunctionload_checkpoint
Load a model checkpoint.
anisoraV1_train_gpu/swissarmytransformer-npu_t_sp/sat/training/model_io.py:260
↓ 7 callersMethodmax
(self)
anisoraV1_infer/sat/examples/yolos/util/misc.py:72
↓ 7 callersFunctionnonlinearity
(x)
anisora_rl/sat/sgm/modules/diffusionmodules/model.py:44
↓ 7 callersFunctionnonlinearity
(x)
anisoraV1_train_npu/sgm/modules/diffusionmodules/model.py:44
↓ 7 callersFunctionnonlinearity
(x)
anisoraV1_train_npu/sat/sgm/modules/diffusionmodules/model.py:44
↓ 7 callersFunctionnonlinearity
(x)
anisoraV1_infer/fastercache/models/cogvideox/sgm/modules/diffusionmodules/model.py:44
↓ 7 callersFunctionnonlinearity
(x)
anisoraV1_train_gpu/sgm/modules/diffusionmodules/model.py:44
↓ 7 callersFunctionnonlinearity
(x)
anisoraV1_train_gpu/sat/sgm/modules/diffusionmodules/model.py:44
↓ 7 callersMethodplay
()
reward/character/samurai/sam2/demo/frontend/src/common/components/video/VideoWorkerBridge.ts:266
↓ 7 callersFunctionretrieve_latents
( encoder_output: torch.Tensor, generator: Optional[torch.Generator] = None, sample_mode: str = "sample" )
anisoraV1_infer/cogvideox/pipeline_cogvideox_image2video-gg_bak.py:162
↓ 7 callersMethodsample
(self, video)
reward/character/samurai/sam2/training/dataset/vos_sampler.py:27
↓ 7 callersFunctionsave_checkpoint
(model, optimizer, rank, output_dir, step, discriminator=False)
anisoraV2_gpu/fastvideo/utils/checkpoint.py:23
↓ 7 callersFunctionsave_checkpoint
(model, optimizer, rank, output_dir, step, discriminator=False, save_optimizer=True)
anisoraV2_npu/fastvideo/utils/checkpoint.py:25
↓ 7 callersMethodscale_model_input
Ensures interchangeability with schedulers that need to scale the denoising model input depending on the current timestep. A
anisoraV1_infer/videosys/schedulers/scheduling_dpm_cogvideox.py:238
← previousnext →601–700 of 18,851, ranked by callers