From e86ad63cf985e06c484d474132cab5b588a857ba Mon Sep 17 00:00:00 2001
From: Phil Wang <lucidrains@gmail.com>
Date: Tue, 19 Jan 2021 11:55:13 -0800
Subject: [PATCH] fix typo

---
 big_sleep/cli.py     | 2 +-
 big_sleep/version.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/big_sleep/cli.py b/big_sleep/cli.py
index cb32dbb..81c2eeb 100644
--- a/big_sleep/cli.py
+++ b/big_sleep/cli.py
@@ -18,7 +18,7 @@ def train(
     imagine = Imagine(
         text,
         lr = lr,
-        image_size = image_size
+        image_size = image_size,
         gradient_accumulate_every = gradient_accumulate_every,
         epochs = epochs,
         iterations = iterations,
diff --git a/big_sleep/version.py b/big_sleep/version.py
index a6587ae..13a85f7 100644
--- a/big_sleep/version.py
+++ b/big_sleep/version.py
@@ -1 +1 @@
-__version__ = '0.2.4'
+__version__ = '0.2.5'