Skip to content

Commit ef9f188

Browse files
merged fp16 and split in onnx transform
Signed-off-by: abhishek-singh591 <[email protected]>
1 parent c78d072 commit ef9f188

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

QEfficient/base/onnx_transforms.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,13 +5,13 @@
55
#
66
# ----------------------------------------------------------------------------
77

8-
from typing import Optional, Tuple
9-
from concurrent.futures import ThreadPoolExecutor
10-
from collections import namedtuple
118
import os
9+
from collections import namedtuple
10+
from concurrent.futures import ThreadPoolExecutor
11+
from typing import Optional, Tuple
1212

1313
import numpy as np
14-
from onnx import ModelProto, external_data_helper, numpy_helper, TensorProto
14+
from onnx import ModelProto, TensorProto, external_data_helper, numpy_helper
1515

1616

1717
class OnnxTransform:

0 commit comments

Comments
 (0)