github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-04 02:26:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

update lightning version (#1954)

Browse Source
This commit is contained in:
Fazzie-Maqianli
2022-11-15 16:57:48 +08:00
committed by GitHub
parent 52c6ad26e0
commit 6bdd0a90ca
13 changed files with 29 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/images/diffusion/scripts/img2img.py
View File

@@ -13,7 +13,7 @@ from torchvision.utils import make_grid
from torch import autocast
from contextlib import nullcontext
import time
from pytorch_lightning import seed_everything
from lightning.pytorch import seed_everything
from ldm.util import instantiate_from_config
from ldm.models.diffusion.ddim import DDIMSampler

2
examples/images/diffusion/scripts/txt2img.py
View File

@@ -10,7 +10,7 @@ from itertools import islice
from einops import rearrange
from torchvision.utils import make_grid
import time
from pytorch_lightning import seed_everything
from lightning.pytorch import seed_everything
from torch import autocast
from contextlib import contextmanager, nullcontext