in consistencydecoder/__init__.py [0:0]
def __call__( self, features: torch.Tensor, schedule=[1.0, 0.5],