src/transformers/models/levit/modeling_levit.py/LevitResidualLayer/__init__
class LevitResidualLayer:    def __init__(self, module, drop_rate):
        super().__init__()
        self.module = module
        self.drop_rate = drop_rate