fix issue with EMA class

lucidrains · Apr 27, 2022 · 177b27e · 177b27e
1 parent 0032a49
commit 177b27e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/nuwa_pytorch/train_vqgan_vae.py b/nuwa_pytorch/train_vqgan_vae.py
@@ -146,7 +146,7 @@ def update(self):
 
         self.update_moving_average(self.ema_model, self.online_model)
 
-    def update_moving_average(ma_model, current_model):
+    def update_moving_average(self, ma_model, current_model):
         def calculate_ema(beta, old, new):
             if not exists(old):
                 return new

diff --git a/setup.py b/setup.py
@@ -4,7 +4,7 @@
   name = 'nuwa-pytorch',
   packages = find_packages(exclude=[]),
   include_package_data = True,
-  version = '0.7.3',
+  version = '0.7.4',
   license='MIT',
   description = 'NÜWA - Pytorch',
   author = 'Phil Wang',