MCPcopy Create free account

hub / github.com/SandAI-org/MAGI-1 / functions

Functions318 in github.com/SandAI-org/MAGI-1

⨍ Functions 318◇ Types & classes 59
Functionmake_viewless_tensor
(inp, requires_grad)
inference/model/dit/dit_module.py:313
Functionmodel_parallel_is_initialized
Check if model and data parallel groups are initialized.
inference/infra/distributed/parallel_state.py:442
Methodnll
(self, sample, dims=[1, 2, 3])
inference/model/vae/vae_module.py:750
Methodno_weight_decay
(self)
inference/model/vae/vae_module.py:512
Methodno_weight_decay
(self)
inference/model/vae/vae_module.py:674
Methodpatch_vae_encode
Encode the input video. Args: x (torch.Tensor): Input video tensor with shape (N, C, T, H, W). sample_poster
inference/pipeline/video_process.py:76
Methodpost_validation
(magi_config)
inference/common/config.py:152
Methodprocess
( self, task_mode: str, config_path: str, image_path: str, text_embedd
comfyui/comfy_nodes.py:161
Functionrange_mod_kernel_fwd
( X, # pointer to the input MAP, # map x index to gating index GATINGS, # pointer to the gating
inference/model/dit/dit_module.py:202
Methodreset
Reset the recorded message and time. This method clears the previous message and time, allowing for a fresh start in recordi
inference/common/timer.py:41
Methodsave
(self, video: torch.Tensor, output_path: str, fps: int)
comfyui/comfy_nodes.py:226
Methodsimple_json_decoder
(dct)
inference/common/config.py:142
Methodspatial_downsample_factor
Property representing the spatial downsample factor. Returns: int: The spatial downsample factor.
inference/model/vae/vae_model.py:38
Methodspatial_downsample_factor
(self)
inference/model/vae/vae_model.py:249
Methodswap_key_value_dict
swap between batches
inference/common/dataclass.py:73
Methodtemporal_downsample_factor
Property representing the temporal downsample factor. Returns: int: The temporal downsample factor.
inference/model/vae/vae_model.py:49
Methodtemporal_downsample_factor
(self)
inference/model/vae/vae_model.py:253
Methodto_json
(self, json_path: str)
inference/common/config.py:164
← previous301–318 of 318, ranked by callers