Adapting txt logging for python 3.8

arthur-qiu · Jun 25, 2023 · a1af4ac · a1af4ac
1 parent 58ddbee
commit a1af4ac
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/sgm/models/diffusion.py b/sgm/models/diffusion.py
@@ -258,14 +258,10 @@ def log_conditionings(self, batch: Dict, n: int) -> Dict:
                         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: