Skip to content

GitLab

"...composable_kernel_onnx.git" did not exist on "28354a0fa374f71ceeb72ddccf09796701981b3c"
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