Mirrors
/
EveryDream2trainer
mirror of https://github.com/victorchall/EveryDream2trainer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix restoring optimizer state typo

This commit is contained in:
Damian Stewart 2023-05-14 11:24:13 +02:00 committed by Victor Hall
parent 003b08992b
commit e14973a9da
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
optimizer/optimizers.py
View File

@ -122,10 +122,10 @@ class EveryDreamOptimizer():
"""
te_optimizer_state_path = os.path.join(ckpt_path, OPTIMIZER_TE_STATE_FILENAME)
unet_optimizer_state_path = os.path.join(ckpt_path, OPTIMIZER_UNET_STATE_FILENAME)
if os.path.exists(te_optimizer_state_path) and self.optimizer_unet is not None:
self._load_optimizer(self.optimizer_unet, te_optimizer_state_path)
if os.path.exists(unet_optimizer_state_path) and self.optimizer_te is not None:
self._load_optimizer(self.optimizer_te, unet_optimizer_state_path)
if os.path.exists(te_optimizer_state_path) and self.optimizer_te is not None:
self._load_optimizer(self.optimizer_te, te_optimizer_state_path)
if os.path.exists(unet_optimizer_state_path) and self.optimizer_unet is not None:
self._load_optimizer(self.optimizer_unet, unet_optimizer_state_path)
def create_optimizers(self, args, text_encoder_params, unet_params):
"""