Method decode

diff2flow/ema.py:232–233

(self, *args, **kwargs)

    
    # adaptations for flow matching
    def decode(self, *args, **kwargs):
        return self.ema_model.decode(*args, **kwargs)
    
    def encode(self, *args, **kwargs):
        return self.ema_model.encode(*args, **kwargs)