diff --git a/comfy/sd1_clip.py b/comfy/sd1_clip.py index 9f0d95b0a..bb240526f 100644 --- a/comfy/sd1_clip.py +++ b/comfy/sd1_clip.py @@ -80,7 +80,7 @@ class SDClipModel(torch.nn.Module, ClipTokenWeightEncoder): "pooled", "hidden" ] - def __init__(self, version="openai/clip-vit-large-patch14", device="cpu", max_length=77, + def __init__(self, device="cpu", max_length=77, freeze=True, layer="last", layer_idx=None, textmodel_json_config=None, dtype=None, model_class=comfy.clip_model.CLIPTextModel, special_tokens={"start": 49406, "end": 49407, "pad": 49407}, layer_norm_hidden_state=True, enable_attention_masks=False, zero_out_masked=False, return_projected_pooled=True, return_attention_masks=False, model_options={}): # clip-vit-base-patch32 diff --git a/comfy/supported_models_base.py b/comfy/supported_models_base.py index 68e895519..e2b70e694 100644 --- a/comfy/supported_models_base.py +++ b/comfy/supported_models_base.py @@ -72,6 +72,7 @@ class BASE: self.unet_config = unet_config.copy() self.sampling_settings = self.sampling_settings.copy() self.latent_format = self.latent_format() + self.optimizations = self.optimizations.copy() for x in self.unet_extra_config: self.unet_config[x] = self.unet_extra_config[x]