From 8cc278447e34a58ab630ad9f190e0fe9c441cd41 Mon Sep 17 00:00:00 2001
From: Phil Wang <lucidrains@gmail.com>
Date: Thu, 2 Jun 2022 11:21:58 -0700
Subject: [PATCH] just cast to right types for blur sigma and kernel size augs

---
 dalle2_pytorch/dalle2_pytorch.py | 2 ++
 dalle2_pytorch/version.py        | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/dalle2_pytorch/dalle2_pytorch.py b/dalle2_pytorch/dalle2_pytorch.py
index 58d6712..952c9f2 100644
--- a/dalle2_pytorch/dalle2_pytorch.py
+++ b/dalle2_pytorch/dalle2_pytorch.py
@@ -1704,10 +1704,12 @@ class LowresConditioner(nn.Module):
 
             # allow for drawing a random sigma between lo and hi float values
             if isinstance(blur_sigma, tuple):
+                blur_sigma = tuple(map(float, blur_sigma))
                 blur_sigma = random.uniform(*blur_sigma)
 
             # allow for drawing a random kernel size between lo and hi int values
             if isinstance(blur_kernel_size, tuple):
+                blur_kernel_size = tuple(map(int, blur_kernel_size))
                 kernel_size_lo, kernel_size_hi = blur_kernel_size
                 blur_kernel_size = random.randrange(kernel_size_lo, kernel_size_hi + 1)
 
diff --git a/dalle2_pytorch/version.py b/dalle2_pytorch/version.py
index 400a104..085bc85 100644
--- a/dalle2_pytorch/version.py
+++ b/dalle2_pytorch/version.py
@@ -1 +1 @@
-__version__ = '0.6.7'
+__version__ = '0.6.8'