Method from_pretrained

models/training_utils.py:117–124

(cls, path, model_cls)

    @classmethod
    def from_pretrained(cls, path, model_cls) -> "EMA":
        _, ema_kwargs = model_cls.load_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)

        ema_model.load_state_dict(ema_kwargs)
        return ema_model

    def save_pretrained(self, path):
        if self.model_cls is None: