fix typo

2023-09-17 20:17:54 +02:00 · 2023-09-17 20:17:54 +02:00 · a68ebe3658
parent 3579bec540
commit a68ebe3658
1 changed files with 2 additions and 2 deletions
--- a/train.py
+++ b/train.py
@ -122,8 +122,8 @@ class EveryDreamTrainingState:
        self.tokenizer = tokenizer
        self.scheduler = scheduler
        self.vae = vae
-        self.unet_ema = unet_ema,
-        self.text_encoder = text_encoder_ema
+        self.unet_ema = unet_ema
+        self.text_encoder_ema = text_encoder_ema


@torch.no_grad()