MCPcopy Create free account

hub / github.com/PKU-YuanGroup/Helios / functions

Functions847 in github.com/PKU-YuanGroup/Helios

⨍ Functions 847◇ Types & classes 124
↓ 1 callersMethodset_epoch
(self, epoch)
helios/dataset/dataloader_dmd.py:229
↓ 1 callersMethodset_lr
(self, optimizer, lr)
eval/utils/third_party/amt/trainers/base_trainer.py:119
↓ 1 callersFunctionsetup_distributed_env
()
tools/offload_data/get_short-latents.py:18
↓ 1 callersFunctionsetup_distributed_env
()
tools/offload_data/get_ode-pairs.py:28
↓ 1 callersFunctionsetup_distributed_env
()
tools/offload_data/get_text-embedding.py:20
↓ 1 callersFunctionsetup_distributed_env
()
tools/offload_data/get_long-latents.py:18
↓ 1 callersMethodstage1_sample
( self, latents: torch.Tensor = None, prompt_embeds: torch.Tensor = None, nega
helios/diffusers_version/pipeline_helios_diffusers.py:513
↓ 1 callersMethodstage1_sample
( self, latents: torch.Tensor = None, prompt_embeds: torch.Tensor = None, nega
helios/pipelines/pipeline_helios_ode.py:482
↓ 1 callersMethodstage1_sample
( self, latents: torch.Tensor = None, prompt_embeds: torch.Tensor = None, nega
helios/pipelines/pipeline_helios.py:483
↓ 1 callersMethodstage2_sample
( self, latents: torch.Tensor = None, pyramid_num_stages: int = None, pyramid_
helios/diffusers_version/pipeline_helios_diffusers.py:624
↓ 1 callersMethodstage2_sample
( self, is_first_section, latents: torch.Tensor = None, stage2_num_stages: int
helios/pipelines/pipeline_helios_ode.py:615
↓ 1 callersMethodstage2_sample
( self, latents: torch.Tensor = None, stage2_num_stages: int = None, stage2_nu
helios/pipelines/pipeline_helios.py:617
↓ 1 callersMethodstep
Predict the sample from the previous timestep by reversing the SDE. This function propagates the diffusion process from the learned m
helios/scheduler/scheduling_helios.py:287
↓ 1 callersMethodstep_dmd
( self, model_output: torch.FloatTensor, timestep: float | torch.FloatTensor = None,
helios/diffusers_version/scheduling_helios_diffusers.py:849
↓ 1 callersMethodstep_euler
( self, model_output: torch.FloatTensor, timestep: float | torch.FloatTensor = None,
helios/diffusers_version/scheduling_helios_diffusers.py:311
↓ 1 callersMethodstep_unipc
( self, model_output: torch.Tensor, timestep: int | torch.Tensor = None, sampl
helios/diffusers_version/scheduling_helios_diffusers.py:727
↓ 1 callersFunctiontest_backward_comparison
()
helios/modules/helios_kernels/triton_rope.py:195
↓ 1 callersFunctiontest_comparison
()
helios/modules/helios_kernels/triton_rope.py:169
↓ 1 callersFunctiontest_in_training_loop_comparison
()
helios/modules/helios_kernels/triton_rope.py:255
↓ 1 callersFunctiontest_zero_error
()
helios/modules/helios_kernels/triton_rope.py:142
↓ 1 callersMethodtokenize
Returns the tokenized representation of given input string(s) Parameters ---------- texts : Union[str, List[str]]
eval/utils/third_party/ViCLIP/viclip_text.py:126
↓ 1 callersMethodvalid_mask
(self, tensor)
eval/utils/third_party/amt/losses/loss.py:72
↓ 1 callersMethodvalid_mask
(self, tensor)
eval/utils/third_party/amt/losses/loss.py:108
↓ 1 callersMethodvfi_score
Calculate video frame interpolation quality score
eval/6_get_drifting_motion_smoothness.py:153
↓ 1 callersMethodvfi_score
Calculate video frame interpolation quality score
eval/2_get_motion_smoothness.py:151
↓ 1 callersFunctionwarmup
预热
tools/others/benchmark/benchmark_patchification_performance.py:194
↓ 1 callersFunctionwhitespace_clean
(text)
eval/utils/third_party/ViCLIP/simple_tokenizer.py:68
↓ 1 callersFunctionwhitespace_clean
(text)
helios/utils/utils_base.py:582
↓ 1 callersFunctionwhitespace_clean
(text)
helios/diffusers_version/pipeline_helios_diffusers.py:99
↓ 1 callersFunctionwhitespace_clean
(text)
helios/pipelines/pipeline_helios_ode.py:105
↓ 1 callersFunctionwhitespace_clean
(text)
helios/pipelines/pipeline_helios.py:106
↓ 1 callersFunctionwriteFloat
(name, data)
eval/utils/third_party/amt/utils/utils.py:274
↓ 1 callersFunctionwriteFlow
(name, flow)
eval/utils/third_party/amt/utils/utils.py:241
FunctionFunctionCorrelation
(tenOne, tenTwo, intStride)
eval/utils/third_party/amt/flow_generation/liteflownet/correlation/correlation.py:427
Method__call__
(self, coords0, coords1)
eval/utils/third_party/amt/networks/blocks/raft.py:171
Method__call__
(self, msg=None, level=logging.INFO, tb_msg=None)
eval/utils/third_party/amt/trainers/logger.py:51
Method__call__
Preprocess inputs to token sequences and return a batch
helios/videoalign/data.py:201
Method__call__
( self, attn: "HeliosAttention", hidden_states: torch.Tensor, encoder_hidden_s
helios/modules/transformer_helios.py:196
Method__call__
( self, attn: "HeliosAttention", hidden_states: torch.Tensor, encoder_hidden_s
helios/diffusers_version/transformer_helios_diffusers.py:110
Method__call__
r""" The call function to the pipeline for generation. Args: prompt (`str` or `list[str]`, *optional*): T
helios/diffusers_version/pipeline_helios_diffusers.py:832
Method__call__
r""" The call function to the pipeline for generation. Args: prompt (`str` or `List[str]`, *optional*): T
helios/pipelines/pipeline_helios_ode.py:927
Method__call__
r""" The call function to the pipeline for generation. Args: prompt (`str` or `List[str]`, *optional*): T
helios/pipelines/pipeline_helios.py:862
Method__getitem__
(self, idx)
tools/others/get_mean_std_size9.py:202
Method__getitem__
(self, idx)
tools/offload_data/get_ode-pairs.py:88
Method__getitem__
(self, idx)
tools/offload_data/get_text-embedding.py:80
Method__getitem__
(self, idx)
eval/utils/third_party/amt/datasets/gopro_datasets.py:125
Method__getitem__
(self, idx)
eval/utils/third_party/amt/datasets/gopro_datasets.py:191
Method__getitem__
(self, idx)
eval/utils/third_party/amt/datasets/vimeo_datasets.py:106
Method__getitem__
(self, idx)
eval/utils/third_party/amt/datasets/vimeo_datasets.py:154
Method__getitem__
(self, idx)
eval/utils/third_party/amt/datasets/adobe_datasets.py:43
Method__getitem__
(self, idx)
helios/dataset/dataloader_history_latents_dist.py:196
Method__getitem__
(self, idx)
helios/dataset/dataloader_mp4_dist.py:355
Method__getitem__
(self, idx)
helios/dataset/dataloader_dmd.py:237
Method__init__
( self, feature_folders, history_sizes=[16, 2, 1], is_keep_x0=True, fo
tools/others/get_mean_std_size9.py:16
Method__init__
(self, filename)
tools/others/benchmark/benchmark_compile_performance.py:26
Method__init__
(self, prompts, uttids)
tools/offload_data/get_ode-pairs.py:81
Method__init__
(self, prompts, uttids)
tools/offload_data/get_text-embedding.py:73
Method__init__
(self, height=384, width=640)
eval/6_get_drifting_motion_smoothness.py:24
Method__init__
(self, config, ckpt, height=384, width=640, device="cuda")
eval/6_get_drifting_motion_smoothness.py:67
Method__init__
(self, height=384, width=640)
eval/2_get_motion_smoothness.py:20
Method__init__
(self, config, ckpt, height=384, width=640, device="cuda")
eval/2_get_motion_smoothness.py:63
Method__init__
(self)
eval/utils/third_party/amt/utils/utils.py:16
Method__init__
(self)
eval/utils/third_party/amt/utils/utils.py:33
Method__init__
(self, dims, divisor=16)
eval/utils/third_party/amt/utils/utils.py:60
Method__init__
(self, corr_radius=3, corr_lvls=4, num_flows=3, channels=[20, 32, 44, 56], skip_channels=20)
eval/utils/third_party/amt/networks/AMT-S.py:11
Method__init__
(self)
eval/utils/third_party/amt/networks/IFRNet.py:13
Method__init__
(self)
eval/utils/third_party/amt/networks/IFRNet.py:29
Method__init__
(self)
eval/utils/third_party/amt/networks/IFRNet.py:44
Method__init__
(self)
eval/utils/third_party/amt/networks/IFRNet.py:59
Method__init__
(self)
eval/utils/third_party/amt/networks/IFRNet.py:74
Method__init__
(self, corr_radius=3, corr_lvls=4, num_flows=5, channels=[84, 96, 112, 128], skip_channels=84)
eval/utils/third_party/amt/networks/AMT-G.py:11
Method__init__
(self, corr_radius=3, corr_lvls=4, num_flows=5, channels=[48, 64, 72, 128], skip_channels=48)
eval/utils/third_party/amt/networks/AMT-L.py:13
Method__init__
(self, in_planes, planes, norm_fn="group", stride=1)
eval/utils/third_party/amt/networks/blocks/feat_enc.py:6
Method__init__
(self, in_planes, planes, norm_fn="group", stride=1)
eval/utils/third_party/amt/networks/blocks/feat_enc.py:63
Method__init__
(self, output_dim=128, norm_fn="batch", dropout=0.0)
eval/utils/third_party/amt/networks/blocks/feat_enc.py:187
Method__init__
(self, output_dim=128, norm_fn="batch", dropout=0.0)
eval/utils/third_party/amt/networks/blocks/feat_enc.py:262
Method__init__
(self, in_channels, side_channels, bias=True)
eval/utils/third_party/amt/networks/blocks/ifrnet.py:20
Method__init__
(self, in_ch, out_ch, skip_ch)
eval/utils/third_party/amt/networks/blocks/ifrnet.py:80
Method__init__
(self, in_ch, out_ch, skip_ch)
eval/utils/third_party/amt/networks/blocks/ifrnet.py:98
Method__init__
(self, in_ch, skip_ch, num_flows=3)
eval/utils/third_party/amt/networks/blocks/multi_flow.py:45
Method__init__
(self, cdim, hidden_dim, flow_dim, corr_dim, fc_dim, corr_levels=4, radius=3, scale_factor=None)
eval/utils/third_party/amt/networks/blocks/raft.py:34
Method__init__
( self, cdim, hidden_dim, flow_dim, corr_dim, corr_dim2,
eval/utils/third_party/amt/networks/blocks/raft.py:85
Method__init__
(self)
eval/utils/third_party/amt/flow_generation/liteflownet/run.py:96
Method__init__
(self, intLevel)
eval/utils/third_party/amt/flow_generation/liteflownet/run.py:154
Method__init__
(self, intLevel)
eval/utils/third_party/amt/flow_generation/liteflownet/run.py:249
Method__init__
(self, intLevel)
eval/utils/third_party/amt/flow_generation/liteflownet/run.py:306
Method__init__
(self)
eval/utils/third_party/amt/flow_generation/liteflownet/correlation/correlation.py:435
Method__init__
(self, config)
eval/utils/third_party/amt/trainers/base_trainer.py:22
Method__init__
(self, common_cfg, tb_cfg=None, wandb_cfg=None, rank=0)
eval/utils/third_party/amt/trainers/logger.py:18
Method__init__
(self, dataset_dir="data/GOPRO", interFrames=7, augment=True)
eval/utils/third_party/amt/datasets/gopro_datasets.py:82
Method__init__
(self, dataset_dir="data/GOPRO", interFrames=7)
eval/utils/third_party/amt/datasets/gopro_datasets.py:160
Method__init__
(self, dataset_dir="data/vimeo_triplet", flow_dir=None, augment=True, crop_size=(224, 224))
eval/utils/third_party/amt/datasets/vimeo_datasets.py:81
Method__init__
(self, dataset_dir="data/vimeo_triplet")
eval/utils/third_party/amt/datasets/vimeo_datasets.py:133
Method__init__
(self, dataset_dir="data/adobe240/test_frames", interFrames=7, augment=True)
eval/utils/third_party/amt/datasets/adobe_datasets.py:25
Method__init__
(self, loss_weight, keys)
eval/utils/third_party/amt/losses/loss.py:33
Method__init__
(self, loss_weight, keys)
eval/utils/third_party/amt/losses/loss.py:43
Method__init__
(self, loss_weight, keys, patch_size=7)
eval/utils/third_party/amt/losses/loss.py:56
Method__init__
(self, loss_weight, keys, patch_size=3)
eval/utils/third_party/amt/losses/loss.py:90
Method__init__
(self, loss_weight, keys, beta=0.3)
eval/utils/third_party/amt/losses/loss.py:128
Method__init__
(self, loss_weight, keys, beta=0.3)
eval/utils/third_party/amt/losses/loss.py:165
← previousnext →501–600 of 847, ranked by callers