r/stable-diffusion-finetune
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

disabled_train is set later

Browse source
This commit is contained in:
rromb 2022-05-31 14:18:01 +02:00
parent 0aa5027fdc
commit 07ebb8fb2a
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
ldm/modules/encoders/modules.py
View file

@ -119,7 +119,7 @@ class FrozenT5Embedder(AbstractEncoder):
def freeze(self): def freeze(self):
self.transformer = self.transformer.eval() self.transformer = self.transformer.eval()
self.train = disabled_train #self.train = disabled_train
for param in self.parameters(): for param in self.parameters():
param.requires_grad = False param.requires_grad = False
@ -148,7 +148,7 @@ class FrozenCLIPEmbedder(AbstractEncoder):
def freeze(self): def freeze(self):
self.transformer = self.transformer.eval() self.transformer = self.transformer.eval()
self.train = disabled_train #self.train = disabled_train
for param in self.parameters(): for param in self.parameters():
param.requires_grad = False param.requires_grad = False