From 9b322ea634e33e9e3f014726bb7fe6e9e114e394 Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Mon, 9 May 2022 19:46:19 -0700 Subject: [PATCH] patch --- dalle2_pytorch/dalle2_pytorch.py | 1 - dalle2_pytorch/train.py | 1 + setup.py | 2 +- 3 files changed, 2 insertions(+), 2 deletions(-) diff --git a/dalle2_pytorch/dalle2_pytorch.py b/dalle2_pytorch/dalle2_pytorch.py index ffcc621..1529eae 100644 --- a/dalle2_pytorch/dalle2_pytorch.py +++ b/dalle2_pytorch/dalle2_pytorch.py @@ -5,7 +5,6 @@ from functools import partial from contextlib import contextmanager from collections import namedtuple from pathlib import Path -import time import torch import torch.nn.functional as F diff --git a/dalle2_pytorch/train.py b/dalle2_pytorch/train.py index d086143..86b73e1 100644 --- a/dalle2_pytorch/train.py +++ b/dalle2_pytorch/train.py @@ -1,3 +1,4 @@ +import time import copy from functools import partial diff --git a/setup.py b/setup.py index 8e03de4..faea49e 100644 --- a/setup.py +++ b/setup.py @@ -10,7 +10,7 @@ setup( 'dream = dalle2_pytorch.cli:dream' ], }, - version = '0.2.3', + version = '0.2.4', license='MIT', description = 'DALL-E 2', author = 'Phil Wang',