wassname/denoising-diffusion-pytorch
1
0
Fork 0
mirror of https://github.com/wassname/denoising-diffusion-pytorch.git synced 2026-09-09 11:21:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Solve issue #26

Browse Source
This commit is contained in:
Arya Aftab
2022-06-29 11:42:54 +04:30
committed by GitHub
parent 8408775cfc
commit d97bc0278c
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
denoising_diffusion_pytorch/denoising_diffusion_pytorch.py
+6 -1
View File
@@ -15,12 +15,17 @@ from torch.optim import Adam
from torchvision import transforms, utils
from PIL import Image
from tqdm import tqdm
from einops import rearrange, reduce
from einops.layers.torch import Rearrange
from ema_pytorch import EMA
import sys
if 'ipykernel' in sys.modules:
from tqdm.notebook import tqdm
else:
from tqdm import tqdm
# helpers functions
def exists(x):