From 6edb1c5dd09a30ae117393f342d4b03f1e5bd0ee Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Wed, 27 Apr 2022 16:40:02 -0700 Subject: [PATCH] fix issue with ema class --- dalle2_pytorch/train.py | 2 +- setup.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/dalle2_pytorch/train.py b/dalle2_pytorch/train.py index c8c2647..c35bfe0 100644 --- a/dalle2_pytorch/train.py +++ b/dalle2_pytorch/train.py @@ -35,7 +35,7 @@ class EMA(nn.Module): 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 index ffd1ac5..5c1dbd3 100644 --- a/setup.py +++ b/setup.py @@ -10,7 +10,7 @@ setup( 'dream = dalle2_pytorch.cli:dream' ], }, - version = '0.0.55', + version = '0.0.56', license='MIT', description = 'DALL-E 2', author = 'Phil Wang',