Method freeze

diff2flow/conditioning/encoders.py:180–183

(self)

        self.uncond = None

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

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