From f657c364af7b36422e529948753533f3a7abfb58 Mon Sep 17 00:00:00 2001 From: manickavela29 Date: Thu, 27 Jun 2024 01:09:07 +0000 Subject: [PATCH] lint fix Signed-off-by: manickavela29 --- egs/librispeech/ASR/zipformer/export-onnx-streaming.py | 2 +- egs/librispeech/ASR/zipformer/export-onnx.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/egs/librispeech/ASR/zipformer/export-onnx-streaming.py b/egs/librispeech/ASR/zipformer/export-onnx-streaming.py index bdecee726..6691c88b7 100755 --- a/egs/librispeech/ASR/zipformer/export-onnx-streaming.py +++ b/egs/librispeech/ASR/zipformer/export-onnx-streaming.py @@ -74,8 +74,8 @@ import onnx import torch import torch.nn as nn from decoder import Decoder -from onnxruntime.quantization import QuantType, quantize_dynamic from onnxconverter_common import float16 +from onnxruntime.quantization import QuantType, quantize_dynamic from scaling_converter import convert_scaled_to_non_scaled from train import add_model_arguments, get_model, get_params from zipformer import Zipformer2 diff --git a/egs/librispeech/ASR/zipformer/export-onnx.py b/egs/librispeech/ASR/zipformer/export-onnx.py index fe49da101..c924be613 100755 --- a/egs/librispeech/ASR/zipformer/export-onnx.py +++ b/egs/librispeech/ASR/zipformer/export-onnx.py @@ -70,8 +70,8 @@ import onnx import torch import torch.nn as nn from decoder import Decoder -from onnxruntime.quantization import QuantType, quantize_dynamic from onnxconverter_common import float16 +from onnxruntime.quantization import QuantType, quantize_dynamic from scaling_converter import convert_scaled_to_non_scaled from train import add_model_arguments, get_model, get_params from zipformer import Zipformer2