wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-21 23:41:28 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

wip make package structure coherent

Browse Source
This commit is contained in:
Benjamin Berman
2023-08-22 11:35:20 -07:00
parent bc4e52f790
commit d25d5e75f0
37 changed files with 216 additions and 196 deletions
Download Patch File Download Diff File Expand all files Collapse all files
comfy/ldm/models/autoencoder.py
+4 -4
View File
@@ -3,11 +3,11 @@ import torch
import torch.nn.functional as F
from contextlib import contextmanager
from comfy.ldm.modules.diffusionmodules.model import Encoder, Decoder
from comfy.ldm.modules.distributions.distributions import DiagonalGaussianDistribution
from ..modules.diffusionmodules.model import Encoder, Decoder
from ..modules.distributions.distributions import DiagonalGaussianDistribution
from comfy.ldm.util import instantiate_from_config
from comfy.ldm.modules.ema import LitEma
from ..util import instantiate_from_config
from ..modules.ema import LitEma
# class AutoencoderKL(pl.LightningModule):
class AutoencoderKL(torch.nn.Module):
comfy/ldm/models/diffusion/ddim.py
+1 -1
View File
@@ -4,7 +4,7 @@ import torch
import numpy as np
from tqdm import tqdm
from comfy.ldm.modules.diffusionmodules.util import make_ddim_sampling_parameters, make_ddim_timesteps, noise_like, extract_into_tensor
from ...modules.diffusionmodules.util import make_ddim_sampling_parameters, make_ddim_timesteps, noise_like, extract_into_tensor
class DDIMSampler(object):
comfy/ldm/models/diffusion/plms.py
+2 -2
View File
@@ -5,8 +5,8 @@ import numpy as np
from tqdm import tqdm
from functools import partial
from comfy.ldm.modules.diffusionmodules.util import make_ddim_sampling_parameters, make_ddim_timesteps, noise_like
from comfy.ldm.models.diffusion.sampling_util import norm_thresholding
from ...modules.diffusionmodules.util import make_ddim_sampling_parameters, make_ddim_timesteps, noise_like
from .sampling_util import norm_thresholding
class PLMSSampler(object):