(self,
scale_func_cfg=None,
pose_encoder_cfg=None,
pose_decoder_cfg=None,
moe_route_loss_weight=1.0,
template_kl_loss_weight=0.0001,
**kwargs)
| 228 | class FineMoGenTransformer(DiffusionTransformer): |
| 229 | |
| 230 | def __init__(self, |
| 231 | scale_func_cfg=None, |
| 232 | pose_encoder_cfg=None, |
| 233 | pose_decoder_cfg=None, |
| 234 | moe_route_loss_weight=1.0, |
| 235 | template_kl_loss_weight=0.0001, |
| 236 | **kwargs): |
| 237 | super().__init__(**kwargs) |
| 238 | self.scale_func_cfg = scale_func_cfg |
| 239 | self.joint_embed = PoseEncoder(**pose_encoder_cfg) |
| 240 | self.out = zero_module(PoseDecoder(**pose_decoder_cfg)) |
| 241 | self.moe_route_loss_weight = moe_route_loss_weight |
| 242 | self.template_kl_loss_weight = template_kl_loss_weight |
| 243 | |
| 244 | def build_temporal_blocks(self, sa_block_cfg, ca_block_cfg, ffn_cfg): |
| 245 | self.temporal_decoder_blocks = nn.ModuleList() |
nothing calls this directly
no test coverage detected