aljaz/DALLE2-pytorch
1
1
Fork 0
mirror of https://github.com/lucidrains/DALLE2-pytorch.git synced 2025-12-19 09:44:19 +01:00
Code Issues Packages Projects Releases Wiki Activity

autoswitch tqdm for notebooks (#171)

Browse Source 
avoids printing the `tqdm` progress bar to a newline in notebooks when detected
This commit is contained in:
zion
2022-06-25 18:37:06 -05:00
committed by GitHub
parent 98f0c17759
commit c453f468b1
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
dalle2_pytorch/dalle2_pytorch.py
View File

@@ -1,6 +1,6 @@
import math import math
import random import random
from tqdm import tqdm from tqdm.auto import tqdm
from functools import partial, wraps from functools import partial, wraps
from contextlib import contextmanager from contextlib import contextmanager
from collections import namedtuple from collections import namedtuple