Method freeze

diff2flow/conditioning/encoders.py:236–239

(self)

        self.uncond = None

    def freeze(self):
        self.transformer = self.transformer.eval()
        for param in self.parameters():
            param.requires_grad = False

    def forward(self, text):
        dev = next(self.parameters()).device