From c453f468b1116354a0ba14e4ad13c99ff1246dae Mon Sep 17 00:00:00 2001
From: zion <51308183+nousr@users.noreply.github.com>
Date: Sat, 25 Jun 2022 18:37:06 -0500
Subject: [PATCH] autoswitch tqdm for notebooks (#171)

avoids printing the `tqdm` progress bar to a newline in notebooks when detected
---
 dalle2_pytorch/dalle2_pytorch.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dalle2_pytorch/dalle2_pytorch.py b/dalle2_pytorch/dalle2_pytorch.py
index fad79c2..919339f 100644
--- a/dalle2_pytorch/dalle2_pytorch.py
+++ b/dalle2_pytorch/dalle2_pytorch.py
@@ -1,6 +1,6 @@
 import math
 import random
-from tqdm import tqdm
+from tqdm.auto import tqdm
 from functools import partial, wraps
 from contextlib import contextmanager
 from collections import namedtuple