Fix "Load Checkpoint with config" node.

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

View File

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