diff --git a/src/diffusers/training_utils.py b/src/diffusers/training_utils.py index 0e0d0ce5b568..d2bf3fe07185 100644 --- a/src/diffusers/training_utils.py +++ b/src/diffusers/training_utils.py @@ -379,7 +379,7 @@ def __init__( @classmethod def from_pretrained(cls, path, model_cls, foreach=False) -> "EMAModel": - _, ema_kwargs = model_cls.load_config(path, return_unused_kwargs=True) + _, ema_kwargs = model_cls.from_config(path, return_unused_kwargs=True) model = model_cls.from_pretrained(path) ema_model = cls(model.parameters(), model_cls=model_cls, model_config=model.config, foreach=foreach)