Method from_config

deepspeed/linear/context_manager.py:72–74

(*args, **kwargs)

            return _model_init(model)

        def from_config(*args, **kwargs):
            model = self._orig_causallm_config(*args, **kwargs)
            return _model_init(model)

        torch.nn.Linear = OptLinearWrapper
        if transformers != None: