Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/JaydenLyh/Reward-Forcing
/ types & classes
Types & classes
123 in github.com/JaydenLyh/Reward-Forcing
⨍
Functions
579
◇
Types & classes
123
↓ 23 callers
Class
CausalConv3d
Causal 3d convolusion.
wan/modules/vae.py:17
↓ 18 callers
Class
MemBlock
demo_utils/taehv.py:25
↓ 14 callers
Class
WanDiffusionWrapper
utils/wan_wrapper.py:116
↓ 10 callers
Class
WanLayerNorm
wan/modules/model.py:89
↓ 9 callers
Class
RMS_norm
wan/modules/vae.py:39
↓ 9 callers
Class
WanTextEncoder
utils/wan_wrapper.py:14
↓ 8 callers
Class
AttentionBlock
Causal self-attention with a single head.
wan/modules/vae.py:223
↓ 8 callers
Class
WanVAEWrapper
utils/wan_wrapper.py:54
↓ 7 callers
Class
EMA_FSDP
utils/distributed.py:91
↓ 7 callers
Class
T5LayerNorm
wan/modules/t5.py:53
↓ 6 callers
Class
ResidualBlock
wan/modules/vae.py:186
↓ 6 callers
Class
ResidualBlock
demo_utils/vae.py:13
↓ 6 callers
Class
Upsample
wan/modules/vae.py:57
↓ 6 callers
Class
WanRMSNorm
wan/modules/model.py:70
↓ 5 callers
Class
LayerNorm
wan/modules/clip.py:47
↓ 5 callers
Class
PromptOutput
wan/utils/prompt_extend.py:101
↓ 5 callers
Class
ShardingLMDBDataset
utils/dataset.py:72
↓ 4 callers
Class
FlowDPMSolverMultistepScheduler
`FlowDPMSolverMultistepScheduler` is a fast dedicated high-order solver for diffusion ODEs. This model inherits from [`SchedulerMixin`] and [
wan/utils/fm_solvers.py:69
↓ 4 callers
Class
FlowUniPCMultistepScheduler
`UniPCMultistepScheduler` is a training-free framework designed for the fast sampling of diffusion models. This model inherits from [`Schedu
wan/utils/fm_solvers_unipc.py:20
↓ 4 callers
Class
HuggingfaceTokenizer
wan/modules/tokenizers.py:37
↓ 4 callers
Class
T5RelativeEmbedding
wan/modules/t5.py:221
↓ 4 callers
Class
TextDataset
utils/dataset.py:12
↓ 3 callers
Class
T5Attention
wan/modules/t5.py:69
↓ 3 callers
Class
TGrow
demo_utils/taehv.py:48
↓ 3 callers
Class
TPool
demo_utils/taehv.py:37
↓ 2 callers
Class
CausVid
model/causvid.py:8
↓ 2 callers
Class
DashScopePromptExpander
wan/utils/prompt_extend.py:157
↓ 2 callers
Class
DataConfig
videoalign/data.py:16
↓ 2 callers
Class
FlowMatchScheduler
utils/scheduler.py:106
↓ 2 callers
Class
MLPProj
wan/modules/model.py:469
↓ 2 callers
Class
ModelConfig
videoalign/utils.py:55
↓ 2 callers
Class
PEFTLoraConfig
videoalign/utils.py:34
↓ 2 callers
Class
QuickGELU
wan/modules/clip.py:41
↓ 2 callers
Class
QwenPromptExpander
wan/utils/prompt_extend.py:300
↓ 2 callers
Class
Resample
wan/modules/vae.py:66
↓ 2 callers
Class
SelfForcingTrainingPipeline
pipeline/self_forcing_training.py:8
↓ 2 callers
Class
SiD
model/sid.py:8
↓ 2 callers
Class
T5EncoderModel
wan/modules/t5.py:472
↓ 2 callers
Class
T5FeedForward
wan/modules/t5.py:123
↓ 2 callers
Class
TrainingConfig
videoalign/utils.py:13
↓ 2 callers
Class
VideoVLMRewardInference
videoalign/wan_inference.py:33
↓ 2 callers
Class
WanVAE
wan/modules/vae.py:639
↓ 1 callers
Class
AttentionBlock
wan/modules/clip.py:112
↓ 1 callers
Class
AttentionBlock
wan/modules/xlm_roberta.py:49
↓ 1 callers
Class
AttentionPool
wan/modules/clip.py:156
↓ 1 callers
Class
CLIPModel
wan/modules/clip.py:501
↓ 1 callers
Class
CausalDiffusion
model/diffusion.py:8
↓ 1 callers
Class
CausalDiffusionInferencePipeline
pipeline/causal_diffusion_inference.py:10
↓ 1 callers
Class
CausalHead
wan/modules/causal_model.py:358
↓ 1 callers
Class
CausalInferencePipeline
pipeline/causal_inference.py:9
↓ 1 callers
Class
CausalWanAttentionBlock
wan/modules/causal_model.py:263
↓ 1 callers
Class
CausalWanSelfAttention
wan/modules/causal_model.py:59
↓ 1 callers
Class
Clamp
demo_utils/taehv.py:20
↓ 1 callers
Class
DMD
model/dmd.py:9
↓ 1 callers
Class
Decoder3d
wan/modules/vae.py:369
↓ 1 callers
Class
Encoder3d
wan/modules/vae.py:265
↓ 1 callers
Class
GAN
model/gan.py:10
↓ 1 callers
Class
GELU
wan/modules/t5.py:46
↓ 1 callers
Class
GanAttentionBlock
wan/modules/model.py:357
↓ 1 callers
Class
Head
wan/modules/model.py:439
↓ 1 callers
Class
ODERegression
model/ode_regression.py:9
↓ 1 callers
Class
ODERegressionLMDBDataset
utils/dataset.py:37
↓ 1 callers
Class
PartialEmbeddingUpdateCallback
Callback to update the embedding of special tokens Only the special tokens are updated, the rest of the embeddings are kept fixed
videoalign/trainer.py:219
↓ 1 callers
Class
QWen2VLDataCollator
videoalign/data.py:131
↓ 1 callers
Class
ReDMD
model/re_dmd.py:8
↓ 1 callers
Class
RegisterTokens
wan/modules/model.py:484
↓ 1 callers
Class
Resample
demo_utils/vae_block3.py:9
↓ 1 callers
Class
Resample
demo_utils/vae.py:51
↓ 1 callers
Class
SelfAttention
wan/modules/clip.py:53
↓ 1 callers
Class
SelfAttention
wan/modules/xlm_roberta.py:10
↓ 1 callers
Class
SwiGLU
wan/modules/clip.py:94
↓ 1 callers
Class
T5CrossAttention
wan/modules/t5.py:178
↓ 1 callers
Class
T5Decoder
wan/modules/t5.py:315
↓ 1 callers
Class
T5Encoder
wan/modules/t5.py:267
↓ 1 callers
Class
T5SelfAttention
wan/modules/t5.py:144
↓ 1 callers
Class
TAEHV
demo_utils/taehv.py:159
↓ 1 callers
Class
TextImagePairDataset
utils/dataset.py:127
↓ 1 callers
Class
VAECalibrator
demo_utils/vae_torch2trt.py:139
↓ 1 callers
Class
VAEDecoder3d
demo_utils/vae_block3.py:187
↓ 1 callers
Class
VAEDecoder3d
demo_utils/vae.py:199
↓ 1 callers
Class
VAEDecoderWrapperSingle
demo_utils/vae.py:151
↓ 1 callers
Class
VideoTensorReader
demo_utils/taehv.py:247
↓ 1 callers
Class
VideoTensorWriter
demo_utils/taehv.py:263
↓ 1 callers
Class
VideoVLMRewardTrainer
videoalign/trainer.py:245
↓ 1 callers
Class
VisionTransformer
wan/modules/clip.py:209
↓ 1 callers
Class
WanAttentionBlock
wan/modules/model.py:275
↓ 1 callers
Class
WanGanCrossAttention
wan/modules/model.py:197
↓ 1 callers
Class
WanSelfAttention
wan/modules/model.py:102
↓ 1 callers
Class
WanVAE_
wan/modules/vae.py:483
↓ 1 callers
Class
XLMRoberta
XLMRobertaModel with no pooler and no LM head.
wan/modules/xlm_roberta.py:76
↓ 1 callers
Class
XLMRobertaWithHead
wan/modules/clip.py:303
Class
BaseModel
model/base.py:12
Class
BaseRLModel
model/base.py:223
Class
BidirectionalDiffusionInferencePipeline
pipeline/bidirectional_diffusion_inference.py:10
Class
BidirectionalInferencePipeline
pipeline/bidirectional_inference.py:7
Class
CausalWanModel
r""" Wan diffusion backbone supporting both text-to-video and image-to-video.
wan/modules/causal_model.py:389
Class
DenoisingLoss
utils/loss.py:5
Class
DynamicSwapInstaller
demo_utils/memory.py:13
Class
FlowPredLoss
utils/loss.py:61
Class
NoisePredLoss
utils/loss.py:50
next →
1–100 of 123, ranked by callers