def from_pretrained()

in src/image_gen_aux/preprocessors/depth/depth_preprocessor.py [0:0]

• 3 lines of code
• 1 McCabe index (conditional complexity)

    def from_pretrained(cls, pretrained_model_or_path: Union[str, os.PathLike], **kwargs):
        model = AutoModelForDepthEstimation.from_pretrained(pretrained_model_or_path, **kwargs)

        return cls(model)