From 44b319cb57ed6a144ebe9a96cdec6aec46b5f4f2 Mon Sep 17 00:00:00 2001 From: Ray Bell Date: Wed, 4 May 2022 10:42:20 -0400 Subject: [PATCH] add missing import (#56) --- dalle2_pytorch/cli.py | 1 + 1 file changed, 1 insertion(+) diff --git a/dalle2_pytorch/cli.py b/dalle2_pytorch/cli.py index aca5267..a2a6650 100644 --- a/dalle2_pytorch/cli.py +++ b/dalle2_pytorch/cli.py @@ -1,6 +1,7 @@ import click import torch import torchvision.transforms as T +from functools import reduce from pathlib import Path from dalle2_pytorch import DALLE2, Decoder, DiffusionPrior