mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-09-14 21:45:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

New CheckpointLoaderSimple to load checkpoints without a config.

Browse Source
This commit is contained in:
comfyanonymous
2023-03-03 03:37:35 -05:00
parent c1f5855ac1
commit 94bb0375b0
3 changed files with 127 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
comfy/ldm/models/diffusion/ddpm.py
View File

@@ -81,7 +81,7 @@ class DDPM(torch.nn.Module):
super().__init__()
assert parameterization in ["eps", "x0", "v"], 'currently only supporting "eps" and "x0" and "v"'
self.parameterization = parameterization
print(f"{self.__class__.__name__}: Running in {self.parameterization}-prediction mode")
# print(f"{self.__class__.__name__}: Running in {self.parameterization}-prediction mode")
self.cond_stage_model = None
self.clip_denoised = clip_denoised
self.log_every_t = log_every_t
@@ -522,8 +522,8 @@ class LatentDiffusion(DDPM):
"""main class"""
def __init__(self,
first_stage_config,
cond_stage_config,
first_stage_config={},
cond_stage_config={},
num_timesteps_cond=None,
cond_stage_key="image",
cond_stage_trainable=False,
@@ -562,8 +562,6 @@ class LatentDiffusion(DDPM):
# self.instantiate_first_stage(first_stage_config)
# self.instantiate_cond_stage(cond_stage_config)
self.first_stage_config = first_stage_config
self.cond_stage_config = cond_stage_config
self.cond_stage_forward = cond_stage_forward
self.clip_denoised = False