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

fix typo

This commit is contained in:
Damian Stewart 2023-09-17 20:17:54 +02:00
parent 3579bec540
commit a68ebe3658
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

@ -122,8 +122,8 @@ class EveryDreamTrainingState:
self.tokenizer = tokenizer self.tokenizer = tokenizer
self.scheduler = scheduler self.scheduler = scheduler
self.vae = vae self.vae = vae
self.unet_ema = unet_ema, self.unet_ema = unet_ema
self.text_encoder = text_encoder_ema self.text_encoder_ema = text_encoder_ema
@torch.no_grad() @torch.no_grad()