Unverified Commit b41ce1e0 authored by rebel-kblee's avatar rebel-kblee Committed by GitHub
Browse files

fix multicontrolnet `save_pretrained` logic for compatibility (#7821)



fix multicontrolnet save_pretrained logic for compatibility
Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
parent 95d37484
...@@ -100,20 +100,16 @@ class MultiControlNetModel(ModelMixin): ...@@ -100,20 +100,16 @@ class MultiControlNetModel(ModelMixin):
variant (`str`, *optional*): variant (`str`, *optional*):
If specified, weights are saved in the format pytorch_model.<variant>.bin. If specified, weights are saved in the format pytorch_model.<variant>.bin.
""" """
idx = 0 for idx, controlnet in enumerate(self.nets):
model_path_to_save = save_directory suffix = "" if idx == 0 else f"_{idx}"
for controlnet in self.nets:
controlnet.save_pretrained( controlnet.save_pretrained(
model_path_to_save, save_directory + suffix,
is_main_process=is_main_process, is_main_process=is_main_process,
save_function=save_function, save_function=save_function,
safe_serialization=safe_serialization, safe_serialization=safe_serialization,
variant=variant, variant=variant,
) )
idx += 1
model_path_to_save = model_path_to_save + f"_{idx}"
@classmethod @classmethod
def from_pretrained(cls, pretrained_model_path: Optional[Union[str, os.PathLike]], **kwargs): def from_pretrained(cls, pretrained_model_path: Optional[Union[str, os.PathLike]], **kwargs):
r""" r"""
......
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