nodes/smthemex-ComfyUI_CustomNet
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update ddpm.py

Browse Source
This commit is contained in:
smthemex
2024-08-11 16:23:01 +08:00
committed by GitHub
parent a91be1870f
commit b50a276b7a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
custom_net/ddpm.py
+1 -1
View File
@@ -20,7 +20,7 @@ from torchvision.utils import make_grid
from pytorch_lightning.utilities.rank_zero import rank_zero_only
from omegaconf import ListConfig
from customnet_util import log_txt_as_img, exists, default, ismap, isimage, mean_flat, count_params, instantiate_from_config
from .customnet_util import log_txt_as_img, exists, default, ismap, isimage, mean_flat, count_params, instantiate_from_config
from .ema import LitEma
from .distributions import normal_kl, DiagonalGaussianDistribution
from .autoencoder import VQModelInterface, IdentityFirstStage, AutoencoderKL