MCPcopy Create free account

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

Functions847 in github.com/PKU-YuanGroup/Helios

⨍ Functions 847◇ Types & classes 124
↓ 447 callersMethodto
r"""Move internal buffers of the ExponentialMovingAverage to `device`. Args: device: like `device` argument to `torch.Tensor.to`
helios/utils/create_ema_zero3.py:300
↓ 228 callersMethodupdate
(self, val, n=1)
eval/utils/third_party/amt/utils/utils.py:25
↓ 49 callersMethodfrom_pretrained
(cls, path, model_cls)
helios/utils/create_ema_zero3.py:124
↓ 45 callersMethodload_state_dict
r""" Args: Loads the ExponentialMovingAverage state. This method is used by accelerate during checkpointing to save the ema st
helios/utils/create_ema_zero3.py:356
↓ 42 callersFunctionresize
(x, scale_factor)
eval/utils/third_party/amt/networks/blocks/ifrnet.py:8
↓ 37 callersFunctionread
(file)
eval/utils/third_party/amt/utils/utils.py:101
↓ 29 callersMethodencode
(self, text)
eval/utils/third_party/ViCLIP/simple_tokenizer.py:136
↓ 26 callersFunctionimg2tensor
(img)
eval/utils/third_party/amt/utils/utils.py:84
↓ 26 callersMethodresize
(self, x, scale_factor)
eval/utils/third_party/amt/losses/loss.py:155
↓ 23 callersMethodpad
(self, *inputs)
eval/utils/third_party/amt/utils/utils.py:66
↓ 22 callersMethodstate_dict
r""" Returns the state of the ExponentialMovingAverage as a dict. This method is used by accelerate during checkpointing to save the e
helios/utils/create_ema_zero3.py:309
↓ 21 callersMethodmove_to_cpu
(self, model, non_blocking=True, offload_grad=False)
helios/utils/utils_helios_post.py:268
↓ 18 callersMethoddecode
(self, tokens)
eval/utils/third_party/ViCLIP/simple_tokenizer.py:144
↓ 18 callersMethodstep
(self, parameters: Iterable[torch.nn.Parameter])
helios/utils/create_ema_zero3.py:226
↓ 17 callersFunctionconvrelu
(in_channels, out_channels, kernel_size=3, stride=1, padding=1, dilation=1, groups=1, bias=True)
eval/utils/third_party/amt/networks/blocks/ifrnet.py:12
↓ 16 callersMethodmove_to_gpu
(self, model, device, non_blocking=True, load_grad=False)
helios/utils/utils_helios_post.py:294
↓ 15 callersFunctionconvert_flow_pred_to_x0
(flow_pred, xt, timestep, sigmas, timesteps)
helios/utils/utils_helios_post.py:617
↓ 14 callersFunctionbuild_from_cfg
(config)
eval/utils/third_party/amt/utils/build_utils.py:14
↓ 14 callersMethodset_timesteps
Setting the timesteps and sigmas for each stage
helios/scheduler/scheduling_helios.py:216
↓ 14 callersFunctionwarp
(img, flow)
eval/utils/third_party/amt/utils/flow_utils.py:10
↓ 14 callersMethodwrite
(self, message)
tools/others/benchmark/benchmark_compile_performance.py:30
↓ 12 callersFunctionadd_noise
(original_samples, noise, timestep, sigmas, timesteps)
helios/utils/utils_helios_post.py:608
↓ 12 callersFunctionapply_schedule_shift
( sigmas, noise, sigmas_two=None, base_seq_len: int = 256, max_seq_len: int = 4096, ba
helios/utils/utils_base.py:531
↓ 12 callersMethodstore
r""" Args: Save the current parameters for restoring later. parameters: Iterable of `torch.nn.Parameter`; the parameters t
helios/utils/create_ema_zero3.py:329
↓ 12 callersFunctionunwrap_model
(model)
train_helios.py:559
↓ 11 callersMethodsave
(self, name, epoch)
eval/utils/third_party/amt/trainers/base_trainer.py:224
↓ 10 callersFunctionalign_dimension
(value, alignment=2)
eval/utils/utils.py:84
↓ 9 callersFunctionget_config_value
(args, name)
helios/utils/utils_base.py:37
↓ 8 callersMethodbackward
(ctx, *grads)
helios/modules/helios_kernels/tiled_linear.py:114
↓ 8 callersFunctioncalculate_psnr
(img1, img2)
eval/utils/third_party/amt/metrics/psnr_ssim.py:135
↓ 8 callersFunctionload_video
(video_path, data_transform=None, num_frames=None, return_tensor=True, width=None, height=None)
eval/utils/utils.py:88
↓ 7 callersMethod__init__
( self, in_features: int, out_features: int, rank: int = 128, device="
helios/modules/transformer_helios.py:130
↓ 7 callersMethod_sigma_to_alpha_sigma_t
(self, sigma)
helios/scheduler/scheduling_helios.py:361
↓ 7 callersMethod_sigma_to_alpha_sigma_t
(self, sigma)
helios/diffusers_version/scheduling_helios_diffusers.py:361
↓ 7 callersFunctioncalculate_ssim
(img1, img2, window_size=11, window=None, size_average=True, full=False, val_range=None)
eval/utils/third_party/amt/metrics/psnr_ssim.py:81
↓ 7 callersFunctionget_corrupt_noise_sigma
(model_input, batch_size, corrupt_ratio=1 / 3, num_frames=None, is_frame_independent=False)
helios/utils/utils_helios_base.py:198
↓ 7 callersMethodgradient_checkpointing_method
(self, block, *args)
helios/modules/transformer_helios.py:1108
↓ 7 callersFunctionload_extra_components
Load patch_short, patch_mid, patch_long, q_loras, k_loras, v_loras into the model
helios/utils/utils_base.py:249
↓ 7 callersFunctionreplace_all_norms_with_flash_norms
(model)
helios/modules/helios_kernels/triton_norm.py:14
↓ 6 callersMethod__init__
mapping: map the kwargs keys into desired ones.
eval/utils/third_party/amt/losses/loss.py:8
↓ 6 callersFunctionapply_rotary_emb_transposed
( hidden_states: torch.Tensor, freqs_cis: torch.Tensor, )
helios/modules/transformer_helios.py:63
↓ 6 callersMethodbackward
(ctx, grad_output)
helios/utils/utils_helios_post.py:362
↓ 6 callersFunctioncal_gan_loss
(logit, label=1)
helios/utils/utils_helios_post.py:370
↓ 6 callersFunctioncorrupt_history_latents
( latents_history_short, latents_history_mid, latents_history_long, latent_window_size, is
helios/utils/utils_helios_base.py:265
↓ 6 callersFunctiondownsample_corrupt
(model_input, downsample_min_corrupt_ratio, downsample_max_corrupt_ratio)
helios/utils/utils_helios_base.py:172
↓ 6 callersFunctionreplace_rope_with_flash_rope
()
helios/modules/helios_kernels/triton_rope.py:15
↓ 6 callersFunctionresize
(x, scale_factor)
eval/utils/third_party/amt/networks/blocks/raft.py:6
↓ 6 callersMethodset_epoch
(self, epoch)
helios/dataset/dataloader_dmd.py:351
↓ 6 callersFunctiontorch_gpu_device
(device)
helios/modules/helios_kernels/utils.py:48
↓ 5 callersMethod__init__
(self)
eval/utils/third_party/amt/networks/IFRNet.py:89
↓ 5 callersMethod__init__
( self, dim: int, heads: int = 8, dim_head: int = 64, eps: float = 1e-
helios/diffusers_version/transformer_helios_diffusers.py:164
↓ 5 callersMethod_get_updateblock
(self, cdim, scale_factor=None)
eval/utils/third_party/amt/networks/AMT-G.py:37
↓ 5 callersFunctionbuild_prompt
(prompt, dimension, template_type)
helios/videoalign/prompt_template.py:103
↓ 5 callersFunctioncupy_kernel
(strFunction, objVariables)
eval/utils/third_party/amt/flow_generation/liteflownet/correlation/correlation.py:201
↓ 5 callersFunctioncupy_launch
(strFunction, strKernel)
eval/utils/third_party/amt/flow_generation/liteflownet/correlation/correlation.py:253
↓ 5 callersFunctionencode_prompt
( tokenizer, text_encoder, prompt: Union[str, List[str]], num_videos_per_prompt: int = 1,
helios/utils/utils_base.py:642
↓ 5 callersFunctiongenerate_and_sync_flag
(accelerator, num_denoising_steps, last_step_only=False, is_sync=True)
helios/utils/utils_helios_post.py:575
↓ 5 callersFunctionget_timestep_grid
Get the grid index for a given timesteps.
helios/utils/utils_recycle_single.py:201
↓ 5 callersFunctionget_timestep_grid
Get the grid index for a given timesteps.
helios/utils/utils_recycle_batch.py:287
↓ 5 callersFunctionreplace_rmsnorm_with_fp32
(model)
helios/modules/helios_kernels/fp32_rmsnorm.py:11
↓ 5 callersFunctionrun_single_config
运行单个配置的完整测试
tools/others/benchmark/benchmark_triton_performance.py:250
↓ 5 callersFunctionsafe_item
(value)
train_helios.py:1103
↓ 4 callersMethod__init__
(self, output_dim=128, norm_fn="batch", dropout=0.0)
eval/utils/third_party/amt/networks/blocks/feat_enc.py:114
↓ 4 callersMethod__init__
(self)
eval/utils/third_party/amt/flow_generation/liteflownet/run.py:92
↓ 4 callersMethod_make_layer
(self, dim, stride=1)
eval/utils/third_party/amt/networks/blocks/feat_enc.py:303
↓ 4 callersFunctionapply_rotary_emb_transposed_flash
(x, freqs_cis)
helios/modules/helios_kernels/triton_rope.py:11
↓ 4 callersFunctionapply_rotary_emb_transposed_orig
(x, freqs_cis)
helios/modules/helios_kernels/triton_rope.py:143
↓ 4 callersMethodbackward
(ctx, grad_output)
helios/modules/helios_kernels/triton_rope.py:110
↓ 4 callersFunctionceil_by_factor
Returns the smallest integer greater than or equal to 'number' that is divisible by 'factor'.
helios/videoalign/vision_process.py:40
↓ 4 callersMethodclear_kv_cache
(self)
helios/modules/transformer_helios.py:1125
↓ 4 callersFunctionclip_transform
(n_px)
eval/utils/utils.py:21
↓ 4 callersFunctioncreate_dummy_inputs
创建transformer的dummy输入
tools/others/benchmark/benchmark_triton_performance.py:41
↓ 4 callersFunctioncreate_dummy_inputs
创建transformer的dummy输入
tools/others/benchmark/benchmark_patchification_performance.py:35
↓ 4 callersMethodencode_text
encode text. Args: text (dict): The output of huggingface's `PreTrainedTokenizer`. contains keys: - input_ids (tor
eval/utils/third_party/ViCLIP/viclip.py:132
↓ 4 callersMethodencode_vision
encode image / videos as features. Args: image (torch.Tensor): The input images. test (bool): Whether testing.
eval/utils/third_party/ViCLIP/viclip.py:110
↓ 4 callersFunctiongather_with_optional_gpu_dim
(tensor, keep_gpu_dim=False)
helios/utils/utils_recycle_batch.py:645
↓ 4 callersFunctionget_saturation
(x1, saturation_ratio_min, saturation_ratio_max)
helios/utils/utils_helios_base.py:483
↓ 4 callersFunctionpad_for_3d_conv
(x, kernel_size)
helios/modules/transformer_helios.py:50
↓ 4 callersFunctionpad_for_3d_conv
(x, kernel_size)
helios/diffusers_version/transformer_helios_diffusers.py:40
↓ 4 callersFunctionreadImage
(name)
eval/utils/third_party/amt/utils/utils.py:225
↓ 4 callersFunctionround_by_factor
Returns the closest integer to 'number' that is divisible by 'factor'.
helios/videoalign/vision_process.py:35
↓ 4 callersMethodsave_pretrained
(self, path)
helios/utils/create_ema_zero3.py:134
↓ 4 callersMethodset_epoch
(self, epoch)
tools/others/get_mean_std_size9.py:136
↓ 4 callersFunctionsmart_resize
Rescales the image so that the following conditions are met: 1. Both dimensions (height and width) are divisible by 'factor'. 2. The to
helios/videoalign/vision_process.py:50
↓ 4 callersFunctionstep_recycle
Args: timestep: scalar, 1D tensor with shape [batch_size], or tensor that can be flattened
helios/utils/utils_recycle_batch.py:206
↓ 4 callersMethodunpad
(self, *inputs)
eval/utils/third_party/amt/utils/utils.py:72
↓ 4 callersFunctionwriteImage
(name, data)
eval/utils/third_party/amt/utils/utils.py:235
↓ 3 callersMethod__init__
(self, channels, large=False)
eval/utils/third_party/amt/networks/blocks/ifrnet.py:58
↓ 3 callersMethod_corr_scale_lookup
(self, corr_fn, coord, flow0, flow1, embt, downsample=1)
eval/utils/third_party/amt/networks/AMT-S.py:49
↓ 3 callersMethod_corr_scale_lookup
(self, corr_fn, coord, flow0, flow1, embt, downsample=1)
eval/utils/third_party/amt/networks/AMT-G.py:50
↓ 3 callersMethod_corr_scale_lookup
(self, corr_fn, coord, flow0, flow1, embt, downsample=1)
eval/utils/third_party/amt/networks/AMT-L.py:50
↓ 3 callersMethod_expand_samples
(self, samples, target_length, data_type)
helios/dataset/dataloader_dmd.py:87
↓ 3 callersMethod_get_freqs_base
(self, dim)
helios/modules/transformer_helios.py:678
↓ 3 callersMethod_get_freqs_base
(self, dim)
helios/diffusers_version/transformer_helios_diffusers.py:336
↓ 3 callersMethod_get_updateblock
(self, cdim, scale_factor=None)
eval/utils/third_party/amt/networks/AMT-S.py:37
↓ 3 callersMethod_get_updateblock
(self, cdim, scale_factor=None)
eval/utils/third_party/amt/networks/AMT-L.py:37
↓ 3 callersMethod_init_samples
(self, folders, data_type)
helios/dataset/dataloader_dmd.py:45
↓ 3 callersMethod_make_layer
(self, dim, stride=1)
eval/utils/third_party/amt/networks/blocks/feat_enc.py:153
↓ 3 callersMethod_make_layer
(self, dim, stride=1)
eval/utils/third_party/amt/networks/blocks/feat_enc.py:227
↓ 3 callersFunction_z3_params_to_fetch
(param_list)
helios/utils/create_ema_zero3.py:27
next →1–100 of 847, ranked by callers