Skip to content

GitLab

You need to sign in or sign up before continuing.
Commit c97be4db authored by comfyanonymous's avatar comfyanonymous
Browse files

Support SD2.1 turbo checkpoint.

parent 6b769bca
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
comfy/supported_models.py
View file @ c97be4db
...@@ -71,6 +71,10 @@ class SD20(supported_models_base.BASE): ...@@ -71,6 +71,10 @@ class SD20(supported_models_base.BASE):
return model_base.ModelType.EPS return model_base.ModelType.EPS
def process_clip_state_dict(self, state_dict): def process_clip_state_dict(self, state_dict):
replace_prefix = {}
replace_prefix["conditioner.embedders.0.model."] = "cond_stage_model.model." #SD2 in sgm format
state_dict = utils.state_dict_prefix_replace(state_dict, replace_prefix)
state_dict = utils.transformers_convert(state_dict, "cond_stage_model.model.", "cond_stage_model.clip_h.transformer.text_model.", 24) state_dict = utils.transformers_convert(state_dict, "cond_stage_model.model.", "cond_stage_model.clip_h.transformer.text_model.", 24)
return state_dict return state_dict
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment