From a1af4ac4f15980bb3a15de48e2f3f6ff975b2401 Mon Sep 17 00:00:00 2001 From: Tim Date: Sun, 25 Jun 2023 16:42:45 -0700 Subject: [PATCH] Adapting txt logging for python 3.8 --- sgm/models/diffusion.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/sgm/models/diffusion.py b/sgm/models/diffusion.py index 1cdbf54..e1f1397 100644 --- a/sgm/models/diffusion.py +++ b/sgm/models/diffusion.py @@ -258,14 +258,10 @@ class DiffusionEngine(pl.LightningModule): xc = log_txt_as_img((image_h, image_w), x, size=image_h // 20) else: raise NotImplementedError() - elif isinstance(x, Union[List, ListConfig]): + elif isinstance(x, (List, ListConfig)): if isinstance(x[0], str): # strings xc = log_txt_as_img((image_h, image_w), x, size=image_h // 20) - elif isinstance(x[0], Union[ListConfig, List]): - # # case: videos processed - x = [xx[0] for xx in x] - xc = log_txt_as_img((image_h, image_w), x, size=image_h // 20) else: raise NotImplementedError() else: