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'