def _save_pretrained()

in optimum/furiosa/modeling_base.py [0:0]

• 7 lines of code
• 2 McCabe index (conditional complexity)

    def _save_pretrained(self, save_directory: Union[str, Path], file_name: Optional[str] = None, **kwargs):
        dst_path = Path(save_directory) / FURIOSA_ENF_FILE_NAME

        if isinstance(self.model, (str, Path)):
            copyfile(self.model, dst_path)
        else:
            with open(dst_path, "wb") as f:
                f.write(self.model)