Method export

lmdeploy/turbomind/model_loader.py:50–58

(self)

        )

    def export(self):
        ckpt = create_checkpoint(
            self.model_path,
            mappings=getattr(self.model, '_loader_mappings', []))
        try:
            self.model.model(Prefix(ckpt))
        finally:
            ckpt.close()
        torch.cuda.empty_cache()

    def export_iter(self):
        ckpt = create_checkpoint(