[diffusion] model: support stable-diffusion-3-medium-diffusers (#19225)
Co-authored-by: zhaochenyang20 <zhaochen20@outlook.com> Co-authored-by: Kangrui Du <kangruidu@gmail.com> Co-authored-by: Xiaole Guo <gxlvera@gmail.com>
This commit is contained in:
co-authored by
zhaochenyang20
Kangrui Du
Xiaole Guo
parent
9e6d1c066e
commit
4df60434d7
@@ -181,9 +181,7 @@ class skip_init_modules:
|
||||
|
||||
def _normalize_component_type(module_type: str) -> str:
|
||||
"""Normalize module types like 'text_encoder_2' -> 'text_encoder'."""
|
||||
if module_type.endswith("_2"):
|
||||
return module_type[:-2]
|
||||
return module_type
|
||||
return re.sub(r"_\d+$", "", module_type)
|
||||
|
||||
|
||||
def _clean_hf_config_inplace(model_config: dict) -> None:
|
||||
|
||||
Reference in New Issue
Block a user