funasr/export/export_model.py
@@ -55,12 +55,15 @@ # export encoder1 self.export_config["model_name"] = "model" model = get_model( models = get_model( model, self.export_config, ) if not isinstance(models, tuple): models = (models,) for i, model in enumerate(models): model.eval() # self._export_onnx(model, verbose, export_dir) if self.onnx: self._export_onnx(model, verbose, export_dir) else:
