MCPcopy Create free account

hub / github.com/ModelTC/LightX2V / functions

Functions8,429 in github.com/ModelTC/LightX2V

↓ 7 callersMethod_emit_queue_metrics_unlocked
(self)
lightx2v/server/task_manager.py:311
↓ 7 callersMethod_ensure_unlocked
(self)
lightx2v/utils/lockable_dict.py:139
↓ 7 callersMethod_format_message
(message, args)
lightx2v/disagg/services/base.py:19
↓ 7 callersMethod_init_ring_metadata
(self)
lightx2v/common/kvcache/rolling.py:45
↓ 7 callersMethod_is_truthy
(self, value: Any, default: bool = False)
lightx2v/disagg/services/controller.py:324
↓ 7 callersFunction_reference
(a, b, a_scales, b_scales, bias=None)
test_cases/test_int8_triton_kernels.py:40
↓ 7 callersMethod_sigma_to_alpha_sigma_t
(self, sigma)
lightx2v/models/schedulers/wan/scheduler.py:127
↓ 7 callersMethod_sigma_to_alpha_sigma_t
(self, sigma)
lightx2v/models/schedulers/cosmos3/scheduler.py:249
↓ 7 callersMethod_to_plain
Recursively convert config containers (e.g. LockableDict) to built-in Python types.
lightx2v/disagg/services/controller.py:738
↓ 7 callersMethod_update_lora
(self, lora_path, strength)
lightx2v/models/networks/ltx2/model.py:130
↓ 7 callersMethod_walk_leaves
Yield (name, leaf) for every MM/LN leaf in the tree.
lightx2v/models/networks/worldmirror/weights/transformer_weights.py:572
↓ 7 callersMethodact_quant_func
(self, x)
lightx2v_platform/ops/mm/cambricon_mlu/q_linear.py:24
↓ 7 callersMethodbackward
(ctx, do_s)
lightx2v/common/ops/attn/kernels/sla_kernel.py:275
↓ 7 callersFunctioncheck_path_exists
Check if a file path exists and raise an error if it doesn't. Args: path: The file path to check Raises: FileNotFoundEr
lightx2v/utils/utils.py:705
↓ 7 callersFunctioncreate_default_tensors
Create default tensors (device tensors and pinned memory tensors) based on the source weight device. Args: base_attrs: [(name, attr_name,
lightx2v/common/ops/utils.py:172
↓ 7 callersFunctiondepad_by_length
在指定维度上去除末尾的 padding 部分。 Args: padded_tensor: 已经经过 padding 的 PyTorch tensor。 depadding_len: 需要从末尾去除的长度。 dim: 需要去除 pad
lightx2v/models/networks/worldmirror/comm/padding.py:75
↓ 7 callersFunctiondetect_quant_scheme
根据模型名字自动检测量化精度 - 如果模型名字包含 "int8" → "int8" - 如果模型名字包含 "fp8" 且设备支持 → "fp8" - 否则返回 None(表示不使用量化)
app/utils/model_utils.py:435
↓ 7 callersMethoddevice
(self)
lightx2v/common/kvcache/base.py:97
↓ 7 callersMethodenable_lora
(self, lora_configs, lora_dynamic_apply=False)
lightx2v/pipeline.py:382
↓ 7 callersFunctionerror
Compute SNR between y_pred(tensor) and y_real(tensor) SNR can be calcualted as following equation: SNR(pred, real) = (pred - real)
lightx2v_kernel/python/lightx2v_kernel/utils.py:38
↓ 7 callersFunctionfp16_clamp
(x)
lightx2v/models/input_encoders/hf/wan/t5/model.py:146
↓ 7 callersFunctionget_config_value
(config, key, default=None)
lightx2v/models/runners/bagel/t2i_utils.py:12
↓ 7 callersMethodget_latent_shape_with_target_hw
(self)
lightx2v/models/runners/wan/wan_runner.py:620
↓ 7 callersMethodget_localhost
(self)
lightx2v/disagg/conn.py:545
↓ 7 callersMethodget_session_id
(self)
lightx2v/disagg/conn.py:548
↓ 7 callersMethodget_vae_parallel
(self)
lightx2v/models/runners/wan/wan_runner.py:173
↓ 7 callersMethodinfer_ffn
(self, weights, x, embed0)
lightx2v/models/networks/wan/infer/causvid/transformer_infer.py:199
↓ 7 callersMethodload
(self)
lightx2v/models/video_encoders/hf/ltx2/model.py:53
↓ 7 callersMethodload_quantized
(self, weight_dict)
lightx2v_platform/ops/mm/template.py:82
↓ 7 callersMethodload_vae_decoder
(self)
lightx2v/models/runners/wan/wan_runner.py:205
↓ 7 callersMethodopt
(self, *args, **kwargs)
lightx2v/disagg/services/base.py:51
↓ 7 callersMethodpost_prompt_enhancer
(self)
lightx2v/models/runners/default_runner.py:471
↓ 7 callersMethodprepare_cos_sin
(self, grid_sizes, freqs)
lightx2v/models/networks/wan/infer/pre_infer.py:55
↓ 7 callersFunctionreduce_mean
(value)
lightx2v_train/lightx2v_train/runtime/distributed.py:68
↓ 7 callersMethodremove
(self, room: int)
lightx2v/disagg/services/transformer.py:929
↓ 7 callersFunctionresolve_bagel_t2i_image_shape
(input_info, config)
lightx2v/models/runners/bagel/t2i_utils.py:34
↓ 7 callersMethodself_attn
( self, weights, packed_query_sequence, query_lens, packed_query_posit
lightx2v/models/networks/bagel/infer/transformer_infer.py:85
↓ 7 callersMethodstate_dict
(self)
lightx2v/models/networks/worldmirror/model.py:386
↓ 6 callersMethod__init__
(self, weight)
tools/convert/quant/quant.py:15
↓ 6 callersMethod__init__
( self, weight_name, create_cuda_buffer=False, create_cpu_buffer=False,
lightx2v/common/ops/norm/rms_norm_weight.py:152
↓ 6 callersMethod__init__
(self, n_f, stride)
lightx2v/models/runners/vsr/utils/TCDecoder.py:59
↓ 6 callersMethod__init__
(self, prefix, mm_type)
lightx2v/models/networks/hunyuan3d/weights/transformer_weights.py:27
↓ 6 callersMethod__init__
(self, block_index, mm_type, lora_path=None)
lightx2v/models/networks/neopp/weights/transformer_weights.py:196
↓ 6 callersMethod__init__
(self, config)
lightx2v/models/networks/bagel/vision.py:86
↓ 6 callersMethod__init__
(self, dim, mid_dim)
lightx2v_train/lightx2v_train/model_zoo/native/wan/modules/clip.py:84
↓ 6 callersMethod_append_prompt_template
(self, base: str, addition: str)
lightx2v/models/runners/cosmos3/cosmos3_runner.py:138
↓ 6 callersFunction_attn_modules
(phase, role)
lightx2v/models/networks/wan/infer/s2v/wan_ops.py:161
↓ 6 callersMethod_audio_prepare_single
(self, audio_path)
lightx2v/models/runners/wan/wan_infinitetalk_runner.py:271
↓ 6 callersMethod_compute_rope
(self, position_ids, inv_freq)
lightx2v/models/runners/neopp/neopp_runner.py:93
↓ 6 callersMethod_condition_to_context_tensor
(self, condition, batch_size)
lightx2v_train/lightx2v_train/model_zoo/wan_t2v.py:376
↓ 6 callersMethod_get_action_mode
(self)
lightx2v/models/runners/cosmos3/cosmos3_runner.py:167
↓ 6 callersMethod_get_action_rotary_pos_embed
(self, video_length, head_dim, rope_dim_list=None)
lightx2v/models/networks/wan/infer/matrix_game3/transformer_infer.py:123
↓ 6 callersFunction_get_excluded_time_stack
()
lightx2v/utils/profiler.py:19
↓ 6 callersMethod_infer_action_attn
(self, phase, q, k, v)
lightx2v/models/networks/wan/infer/matrix_game2/transformer_infer.py:59
↓ 6 callersMethod_infer_cond_uncond
(self, inputs, pre_infer_out, infer_condition: bool)
lightx2v/models/networks/neopp/model.py:153
↓ 6 callersMethod_infer_cond_uncond
(self, latents_input, prompt_embeds, infer_condition=True, txt_ids=None, img_ids=None)
lightx2v/models/networks/flux2/model.py:43
↓ 6 callersMethod_init_offload_manager
(self)
lightx2v/models/networks/wan/vace_model.py:27
↓ 6 callersMethod_init_ring_metadata
(self)
lightx2v/common/kvcache/quant.py:153
↓ 6 callersFunction_map_resnet_tail
(tail)
lightx2v/models/video_encoders/hf/wan/vae_2_2.py:37
↓ 6 callersMethod_reshape_heads
(self, x)
lightx2v/models/networks/wan/infer/fastwam/transformer_infer.py:38
↓ 6 callersMethod_send_recv_tensor
发送/接收张量,根据是否使用 FP8 选择通信方式 参数: tensor: 要发送的张量 hidden_dims: 隐藏维度大小 comm: 通信器对象 use_fp8
lightx2v/common/ops/attn/ring_attn.py:369
↓ 6 callersFunction_to_tuple
(x, dim=2)
lightx2v/models/networks/wan/infer/matrix_game2/posemb_layers.py:6
↓ 6 callersFunction_to_tuple
(x, dim=2)
lightx2v/models/networks/hunyuan_video/infer/posemb_layers.py:8
↓ 6 callersMethod_use_mxfp8_quant_fuse
(self)
lightx2v/models/networks/wan/infer/mxfp8_fuse.py:46
↓ 6 callersMethod_write_local_u64
(self, buf: bytearray, value: int)
lightx2v/disagg/rdma_buffer.py:125
↓ 6 callersMethodadd_noise
(self, latent, noise, sigmas)
lightx2v_train/lightx2v_train/schedulers/dmd_scheduler.py:91
↓ 6 callersMethodapply
(self, input_tensor)
lightx2v_kernel/test/mxfp6_mxfp8/test_bench.py:13
↓ 6 callersMethodapply
(self, input_tensor)
lightx2v_kernel/test/mxfp8_mxfp8/test_bench.py:13
↓ 6 callersMethodapply
(self, input_tensor)
lightx2v_kernel/test/mxfp4_mxfp4/test_bench.py:13
↓ 6 callersMethodapply
(self, input_tensor)
lightx2v_kernel/test/nvfp4_nvfp4/test_bench2.py:15
↓ 6 callersMethodapply_lora
(self, lora_configs, model_type=None)
lightx2v/models/networks/lora_adapter.py:24
↓ 6 callersFunctionblock_map_ordinal_lut_triton
(block_map)
lightx2v/common/ops/attn/utils/sparge_util.py:110
↓ 6 callersFunctioncausal_norm_wrapper
(norm_layer: nn.Module, x: torch.Tensor)
lightx2v/models/video_encoders/hf/seedvr/causal_inflation_lib.py:297
↓ 6 callersMethodcfg_cache_name
(cache_name, infer_condition)
lightx2v/models/networks/wan/dreamzero_model.py:29
↓ 6 callersMethodclear_cache
(self)
lightx2v/models/runners/vsr/utils/utils.py:118
↓ 6 callersMethodconnect_to_server
(self, server_ip="127.0.0.1", port=5566)
lightx2v/disagg/rdma_client.py:131
↓ 6 callersFunctioncutlass_scaled_mxfp6_mxfp8_mm
(mat_a, mat_b, scales_a, scales_b, alpha, bias=None)
lightx2v_kernel/python/lightx2v_kernel/gemm.py:150
↓ 6 callersFunctiondefault
(val, d)
lightx2v/models/networks/seedvr/utils/rotary_embedding_torch.py:19
↓ 6 callersMethoddenoise
(self, denoiser_input, timesteps, condition)
lightx2v_train/lightx2v_train/model_zoo/base.py:157
↓ 6 callersMethoddevice_synchronize
( self, )
lightx2v/models/video_encoders/hf/wan/vae.py:1616
↓ 6 callersFunctiondraw_handpose
Draw keypoints and connections representing hand pose on a given canvas. Args: canvas (np.ndarray): A 3D numpy array representing th
tools/preprocess/human_visualization.py:19
↓ 6 callersMethodencode_prompt
(self, prompt)
lightx2v/models/runners/wan/fastwam_runner.py:256
↓ 6 callersFunctionensure_lightx2v_pipeline_stub
()
test_cases/test_wan_mxfp8_fuse_forwarding.py:14
↓ 6 callersFunctionensure_local_lightx2v_kernel
()
test_cases/test_wan_mxfp8_fuse_forwarding.py:21
↓ 6 callersFunctionestimate_transformer_buffer_sizes
(config: Dict[str, Any])
lightx2v/disagg/utils.py:473
↓ 6 callersFunctionexpand_dims
(tensor: torch.Tensor, ndim: int)
lightx2v/models/schedulers/seedvr/utils.py:22
↓ 6 callersMethodfinish
(self)
lightx2v/utils/async_vae.py:162
↓ 6 callersMethodforward_inference
( self, packed_query_sequence: torch.Tensor, query_lens: torch.Tensor, packed_
lightx2v/models/networks/bagel/model.py:178
↓ 6 callersMethodfrom_config
(cls, config: dict)
lightx2v/models/video_encoders/hf/ltx2/upsampler/model.py:18
↓ 6 callersMethodgenerate
(self)
lightx2v/shot_runner/shot_base.py:134
↓ 6 callersMethodget_device
()
lightx2v_platform/base/nvidia.py:30
↓ 6 callersFunctionget_handpose_meta
(keypoints, delta, src_H, src_W)
tools/preprocess/retarget_pose.py:87
↓ 6 callersFunctionget_length
(skeleton, limb)
tools/preprocess/retarget_pose.py:69
↓ 6 callersFunctionget_neighbor
根据指定轴进行位移并保持维度一致
lightx2v/models/video_encoders/hf/hunyuan3d/autoencoders/volume_decoders.py:40
↓ 6 callersFunctionget_padding
(kernel_size: int, dilation: int = 1)
lightx2v/models/video_encoders/hf/ltx2/audio_vae/vocoder.py:19
↓ 6 callersFunctionget_rank
()
lightx2v_train/lightx2v_train/runtime/distributed.py:38
↓ 6 callersFunctionhalf
(x)
lightx2v/models/networks/wan/infer/s2v/wan_ops.py:99
↓ 6 callersMethodimg_emb
(self, weights, x)
lightx2v/models/networks/wan/infer/matrix_game2/pre_infer.py:47
↓ 6 callersMethodinfer_block
(self, weights, grid_sizes, embed, x, embed0, seq_lens, freqs, context, block_idx)
lightx2v/models/networks/wan/infer/feature_caching/transformer_infer.py:926
↓ 6 callersMethodinfer_cross_attn
(self, weights, x, context, block_idx)
lightx2v/models/networks/wan/infer/causvid/transformer_infer.py:141
← previousnext →301–400 of 8,429, ranked by callers