From 07ebb8fb2ae6d713deab7dc53b790184a1e330be Mon Sep 17 00:00:00 2001 From: rromb Date: Tue, 31 May 2022 14:18:01 +0200 Subject: [PATCH] disabled_train is set later --- ldm/modules/encoders/modules.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ldm/modules/encoders/modules.py b/ldm/modules/encoders/modules.py index d5133ab..a62b8e7 100644 --- a/ldm/modules/encoders/modules.py +++ b/ldm/modules/encoders/modules.py @@ -119,7 +119,7 @@ class FrozenT5Embedder(AbstractEncoder): def freeze(self): self.transformer = self.transformer.eval() - self.train = disabled_train + #self.train = disabled_train for param in self.parameters(): param.requires_grad = False @@ -148,7 +148,7 @@ class FrozenCLIPEmbedder(AbstractEncoder): def freeze(self): self.transformer = self.transformer.eval() - self.train = disabled_train + #self.train = disabled_train for param in self.parameters(): param.requires_grad = False