298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-03-15 14:09:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix "Load Checkpoint with config" node.

Browse Source
This commit is contained in:
comfyanonymous 2023-08-29 23:58:32 -04:00
parent d70b0bc43c
commit fe4c07400c
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/sd.py
View File

@ -22,6 +22,7 @@ from . import sdxl_clip
import comfy.model_patcher import comfy.model_patcher
import comfy.lora import comfy.lora
import comfy.t2i_adapter.adapter import comfy.t2i_adapter.adapter
import comfy.supported_models_base
def load_model_weights(model, sd): def load_model_weights(model, sd):
m, u = model.load_state_dict(sd, strict=False) m, u = model.load_state_dict(sd, strict=False)
@ -348,10 +349,11 @@ def load_checkpoint(config_path=None, ckpt_path=None, output_vae=True, output_cl
class EmptyClass: class EmptyClass:
pass pass
model_config = EmptyClass() model_config = comfy.supported_models_base.BASE({})
model_config.unet_config = unet_config
from . import latent_formats from . import latent_formats
model_config.latent_format = latent_formats.SD15(scale_factor=scale_factor) model_config.latent_format = latent_formats.SD15(scale_factor=scale_factor)
model_config.unet_config = unet_config
if config['model']["target"].endswith("LatentInpaintDiffusion"): if config['model']["target"].endswith("LatentInpaintDiffusion"):
model = model_base.SDInpaint(model_config, model_type=model_type) model = model_base.SDInpaint(model_config, model_type=model_type)

2
comfy/supported_models_base.py
View File

@ -1,6 +1,7 @@
import torch import torch
from . import model_base from . import model_base
from . import utils from . import utils
from . import latent_formats
def state_dict_key_replace(state_dict, keys_to_replace): def state_dict_key_replace(state_dict, keys_to_replace):
@ -34,6 +35,7 @@ class BASE:
clip_vision_prefix = None clip_vision_prefix = None
noise_aug_config = None noise_aug_config = None
beta_schedule = "linear" beta_schedule = "linear"
latent_format = latent_formats.LatentFormat
@classmethod @classmethod
def matches(s, unet_config): def matches(s, unet_config):