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

add missing import (#56)

Browse Source
This commit is contained in:
Ray Bell
2022-05-04 10:42:20 -04:00
committed by GitHub
parent c30f380689
commit 44b319cb57
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
dalle2_pytorch/cli.py
View File

@@ -1,6 +1,7 @@
import click import click
import torch import torch
import torchvision.transforms as T import torchvision.transforms as T
from functools import reduce
from pathlib import Path from pathlib import Path
from dalle2_pytorch import DALLE2, Decoder, DiffusionPrior from dalle2_pytorch import DALLE2, Decoder, DiffusionPrior