From a68ebe365803d19fecabb3ea9ba08c2b194c5d9b Mon Sep 17 00:00:00 2001
From: Damian Stewart <d@damianstewart.com>
Date: Sun, 17 Sep 2023 20:17:54 +0200
Subject: [PATCH] fix typo

---
 train.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/train.py b/train.py
index 2e4fbed..59d7920 100644
--- 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()