JaehaerysNL/StableDiffusion
1
0
Fork 0
mirror of https://github.com/Stability-AI/stablediffusion.git synced 2024-12-22 15:44:58 +00:00
Code Issues Projects Releases Packages Wiki Activity

Fix import error in newer version of lighting.

Browse source 
Newer lighting versions move `rank_zero_only` from the distributed
subpackage back to utilities. This commit adds a try-except to catch
that import error, and retry with the alternative.
This commit is contained in:
Ali Hassani 2023-06-15 16:34:44 -07:00
parent cf1d67a6fd
commit eeeb184071
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
ldm/models/diffusion/ddpm.py
View file

@ -17,7 +17,10 @@ from functools import partial
import itertools
from tqdm import tqdm
from torchvision.utils import make_grid
from pytorch_lightning.utilities.distributed import rank_zero_only
try:
from pytorch_lightning.utilities import rank_zero_only
except:
from pytorch_lightning.utilities.distributed import rank_zero_only
from omegaconf import ListConfig
from ldm.util import log_txt_as_img, exists, default, ismap, isimage, mean_flat, count_params, instantiate_from_config