MCPcopy Create free account

hub / github.com/PKU-YuanGroup/Helios / types & classes

Types & classes124 in github.com/PKU-YuanGroup/Helios

⨍ Functions 847◇ Types & classes 124
↓ 7 callersClassResBlock
eval/utils/third_party/amt/networks/blocks/ifrnet.py:19
↓ 6 callersClassIntermediateDecoder
eval/utils/third_party/amt/networks/blocks/ifrnet.py:97
↓ 4 callersClassBucketedFeatureDataset
helios/dataset/dataloader_mp4_dist.py:140
↓ 4 callersClassBucketedSampler
helios/dataset/dataloader_mp4_dist.py:424
↓ 4 callersClassFeedForward
r""" A feed-forward layer. Parameters: dim (`int`): The number of channels in the input. dim_out (`int`, *optional*): The num
helios/modules/helios_kernels/tiled_linear.py:211
↓ 4 callersClassHeliosScheduler
Euler scheduler. This model inherits from [`SchedulerMixin`] and [`ConfigMixin`]. Check the superclass documentation for the generic met
helios/scheduler/scheduling_helios.py:30
↓ 4 callersClassInputPadder
Pads images such that dimensions are divisible by divisor
eval/utils/third_party/amt/utils/utils.py:57
↓ 4 callersClassResidualBlock
eval/utils/third_party/amt/networks/blocks/feat_enc.py:62
↓ 3 callersClassBidirCorrBlock
eval/utils/third_party/amt/networks/blocks/raft.py:148
↓ 3 callersClassCLIP_TEXT
eval/utils/third_party/ViCLIP/viclip_text.py:83
↓ 3 callersClassEncoder
eval/utils/third_party/amt/networks/blocks/ifrnet.py:57
↓ 3 callersClassHeliosAttnProcessor
helios/modules/transformer_helios.py:172
↓ 3 callersClassHeliosSchedulerOutput
helios/diffusers_version/scheduling_helios_diffusers.py:28
↓ 3 callersClassInitDecoder
eval/utils/third_party/amt/networks/blocks/ifrnet.py:79
↓ 3 callersClassLayerNorm
Subclass torch's LayerNorm to handle fp16.
eval/utils/third_party/ViCLIP/viclip_text.py:24
↓ 3 callersClassLoRALinearLayer
helios/modules/transformer_helios.py:129
↓ 3 callersClassMultiFlowDecoder
eval/utils/third_party/amt/networks/blocks/multi_flow.py:44
↓ 3 callersClassVisionTransformer
eval/utils/third_party/ViCLIP/viclip_vision.py:104
↓ 2 callersClassAdaCharbonnierLoss
eval/utils/third_party/amt/losses/loss.py:42
↓ 2 callersClassAdaptiveAntiDrifting
helios/utils/utils_base.py:673
↓ 2 callersClassAverageMeterGroups
eval/utils/third_party/amt/utils/utils.py:32
↓ 2 callersClassBasicUpdateBlock
eval/utils/third_party/amt/networks/blocks/raft.py:84
↓ 2 callersClassBottleneckBlock
eval/utils/third_party/amt/networks/blocks/feat_enc.py:5
↓ 2 callersClassBucketedFeatureDataset
helios/dataset/dataloader_dmd.py:11
↓ 2 callersClassBucketedSampler
helios/dataset/dataloader_history_latents_dist.py:269
↓ 2 callersClassBucketedSampler
helios/dataset/dataloader_dmd.py:324
↓ 2 callersClassDiscriminator3DHead
helios/modules/transformer_helios.py:95
↓ 2 callersClassHeliosAttention
helios/modules/transformer_helios.py:458
↓ 2 callersClassHeliosAttention
helios/diffusers_version/transformer_helios_diffusers.py:160
↓ 2 callersClassHeliosAttnProcessor
helios/diffusers_version/transformer_helios_diffusers.py:100
↓ 2 callersClassHeliosPipelineOutput
r""" Output class for Helios pipelines. Args: frames (`torch.Tensor`, `np.ndarray`, or List[List[PIL.Image.Image]]): List
helios/pipelines/pipeline_output.py:9
↓ 2 callersClassHeliosSchedulerOutput
Output class for the scheduler's `step` function output. Args: prev_sample (`torch.FloatTensor` of shape `(batch_size, num_channels,
helios/scheduler/scheduling_helios.py:14
↓ 2 callersClassSimpleModel
helios/modules/helios_kernels/triton_rope.py:264
↓ 2 callersClassSimpleTokenizer
eval/utils/third_party/ViCLIP/simple_tokenizer.py:74
↓ 2 callersClassViCLIP
docstring for ViCLIP
eval/utils/third_party/ViCLIP/viclip.py:15
↓ 2 callersClassVideoVLMRewardInference
helios/videoalign/inference.py:34
↓ 1 callersClassAdobe240_Dataset
eval/utils/third_party/amt/datasets/adobe_datasets.py:24
↓ 1 callersClassAverageMeter
eval/utils/third_party/amt/utils/utils.py:15
↓ 1 callersClassBasicEncoder
eval/utils/third_party/amt/networks/blocks/feat_enc.py:186
↓ 1 callersClassBucketedFeatureDataset
tools/others/get_mean_std_size9.py:15
↓ 1 callersClassBucketedFeatureDataset
helios/dataset/dataloader_history_latents_dist.py:11
↓ 1 callersClassCustomLogger
eval/utils/third_party/amt/trainers/logger.py:17
↓ 1 callersClassDataConfig
helios/videoalign/data.py:11
↓ 1 callersClassDecoder1
eval/utils/third_party/amt/networks/IFRNet.py:73
↓ 1 callersClassDecoder2
eval/utils/third_party/amt/networks/IFRNet.py:58
↓ 1 callersClassDecoder3
eval/utils/third_party/amt/networks/IFRNet.py:43
↓ 1 callersClassDecoder4
eval/utils/third_party/amt/networks/IFRNet.py:28
↓ 1 callersClassDualLogger
同时输出到控制台和文件的日志器
tools/others/benchmark/benchmark_compile_performance.py:23
↓ 1 callersClassEMAModel_Zero3_LoRA
Exponential Moving Average of models weights
helios/utils/create_ema_zero3_lora.py:23
↓ 1 callersClassEncoder
eval/utils/third_party/amt/networks/IFRNet.py:12
↓ 1 callersClassFeatures
eval/utils/third_party/amt/flow_generation/liteflownet/run.py:95
↓ 1 callersClassFrameProcess
eval/6_get_drifting_motion_smoothness.py:23
↓ 1 callersClassFrameProcess
eval/2_get_motion_smoothness.py:19
↓ 1 callersClassGoPro_Test_Dataset
eval/utils/third_party/amt/datasets/gopro_datasets.py:159
↓ 1 callersClassHeliosOutputNorm
helios/modules/transformer_helios.py:157
↓ 1 callersClassHeliosOutputNorm
helios/diffusers_version/transformer_helios_diffusers.py:85
↓ 1 callersClassHeliosRotaryPosEmbed
helios/modules/transformer_helios.py:669
↓ 1 callersClassHeliosRotaryPosEmbed
helios/diffusers_version/transformer_helios_diffusers.py:327
↓ 1 callersClassHeliosTimeTextEmbedding
helios/modules/transformer_helios.py:623
↓ 1 callersClassHeliosTimeTextEmbedding
helios/diffusers_version/transformer_helios_diffusers.py:291
↓ 1 callersClassHeliosTransformerBlock
helios/modules/transformer_helios.py:718
↓ 1 callersClassHeliosTransformerBlock
helios/diffusers_version/transformer_helios_diffusers.py:376
↓ 1 callersClassLargeEncoder
eval/utils/third_party/amt/networks/blocks/feat_enc.py:261
↓ 1 callersClassMatching
eval/utils/third_party/amt/flow_generation/liteflownet/run.py:153
↓ 1 callersClassModelConfig
helios/videoalign/utils.py:59
↓ 1 callersClassMotionSmoothness
eval/2_get_motion_smoothness.py:62
↓ 1 callersClassMotionSmoothnessEvaluator
eval/6_get_drifting_motion_smoothness.py:66
↓ 1 callersClassNetwork
eval/utils/third_party/amt/flow_generation/liteflownet/run.py:91
↓ 1 callersClassOptimizedLowVRAMManager
helios/utils/utils_helios_post.py:263
↓ 1 callersClassPEFTLoraConfig
helios/videoalign/utils.py:37
↓ 1 callersClassPromptDataset
tools/offload_data/get_ode-pairs.py:80
↓ 1 callersClassPromptDataset
tools/offload_data/get_text-embedding.py:72
↓ 1 callersClassQuickGELU
eval/utils/third_party/ViCLIP/viclip_text.py:33
↓ 1 callersClassQuickGELU
eval/utils/third_party/ViCLIP/viclip_vision.py:49
↓ 1 callersClassRegularization
eval/utils/third_party/amt/flow_generation/liteflownet/run.py:305
↓ 1 callersClassResidualAttentionBlock
eval/utils/third_party/ViCLIP/viclip_text.py:38
↓ 1 callersClassResidualAttentionBlock
eval/utils/third_party/ViCLIP/viclip_vision.py:54
↓ 1 callersClassSmallEncoder
eval/utils/third_party/amt/networks/blocks/feat_enc.py:113
↓ 1 callersClassSmallUpdateBlock
eval/utils/third_party/amt/networks/blocks/raft.py:33
↓ 1 callersClassSubpixel
eval/utils/third_party/amt/flow_generation/liteflownet/run.py:248
↓ 1 callersClassTrainingConfig
helios/videoalign/utils.py:15
↓ 1 callersClassTransformer
eval/utils/third_party/ViCLIP/viclip_text.py:66
↓ 1 callersClassTransformer
eval/utils/third_party/ViCLIP/viclip_vision.py:86
ClassArgs
helios/utils/train_config.py:435
ClassCharbonnierLoss
eval/utils/third_party/amt/losses/loss.py:32
ClassDataConfig
helios/utils/train_config.py:16
ClassEMAModel_Zero3
Exponential Moving Average of models weights
helios/utils/create_ema_zero3.py:32
ClassFP32RMSNorm
helios/modules/helios_kernels/fp32_rmsnorm.py:28
ClassFlash_Layernorm
helios/modules/helios_kernels/triton_norm.py:116
ClassFlash_RMS_Layernorm
helios/modules/helios_kernels/triton_norm.py:329
ClassFlash_RoPE_Transposed
helios/modules/helios_kernels/triton_rope.py:64
ClassGan_D_Loss_With_Cached_Grad
helios/utils/utils_helios_post.py:309
ClassGeometryLoss
eval/utils/third_party/amt/losses/loss.py:89
ClassGoPro_Train_Dataset
eval/utils/third_party/amt/datasets/gopro_datasets.py:81
ClassHeliosAttnProcessor2_0
helios/modules/transformer_helios.py:448
ClassHeliosPipeline
r""" Pipeline for text-to-video / image-to-video / video-to-video generation using Helios. This model inherits from [`DiffusionPipeline`]. Ch
helios/diffusers_version/pipeline_helios_diffusers.py:124
ClassHeliosPipeline
r""" Pipeline for text-to-video / image-to-video / video-to-video generation using Helios. This model inherits from [`DiffusionPipeline`]. Ch
helios/pipelines/pipeline_helios_ode.py:121
ClassHeliosPipeline
r""" Pipeline for text-to-video / image-to-video / video-to-video generation using Helios. This model inherits from [`DiffusionPipeline`]. Ch
helios/pipelines/pipeline_helios.py:122
ClassHeliosScheduler
helios/diffusers_version/scheduling_helios_diffusers.py:35
ClassHeliosTransformer3DModel
r""" A Transformer model for video-like data used in the Helios model. Args: patch_size (`Tuple[int]`, defaults to `(1, 2, 2)`):
helios/modules/transformer_helios.py:905
next →1–100 of 124, ranked by callers