tanner/min-dalle-test
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

delete params sooner

Browse Source
This commit is contained in:
Brett Kuprel 2022-06-30 11:44:36 -04:00
parent f951424e38
commit c2a3858c96
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
min_dalle/min_dalle_torch.py
View File

@ -45,8 +45,8 @@ class MinDalleTorch(MinDalleBase):
is_encoder=True is_encoder=True
) )
self.encoder.load_state_dict(params, strict=False) self.encoder.load_state_dict(params, strict=False)
if torch.cuda.is_available(): self.encoder = self.encoder.cuda()
del params del params
if torch.cuda.is_available(): self.encoder = self.encoder.cuda()
def init_decoder(self): def init_decoder(self):
@ -69,8 +69,8 @@ class MinDalleTorch(MinDalleBase):
is_encoder=False is_encoder=False
) )
self.decoder.load_state_dict(params, strict=False) self.decoder.load_state_dict(params, strict=False)
if torch.cuda.is_available(): self.decoder = self.decoder.cuda()
del params del params
if torch.cuda.is_available(): self.decoder = self.decoder.cuda()
def init_detokenizer(self): def init_detokenizer(self):