# Adapted from Open-Sora-Plan # This source code is licensed under the license found in the # LICENSE file in the root directory of this source tree. # -------------------------------------------------------- # References: # Open-Sora-Plan: https://github.com/PKU-YuanGroup/Open-Sora-Plan # -------------------------------------------------------- import torch.nn as nn class Block(nn.Module): def __init__(self, *args, **kwargs) -> None: super().__init__(*args, **kwargs)