We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent de34841 commit a505dc6Copy full SHA for a505dc6
QEfficient/base/onnx_transforms.py
@@ -6,11 +6,11 @@
6
# ----------------------------------------------------------------------------
7
8
import os
9
-import numpy as np
10
-from typing import Optional, Tuple
11
from concurrent.futures import ThreadPoolExecutor
12
-from onnx import ModelProto, external_data_helper, numpy_helper
+from typing import Optional, Tuple
13
+import numpy as np
+from onnx import ModelProto, external_data_helper, numpy_helper
14
15
16
class OnnxTransform:
@@ -91,4 +91,4 @@ def process_tensor(args):
91
92
with ThreadPoolExecutor(max_workers=os.cpu_count() * 4) as executor:
93
list(executor.map(process_tensor, file_assignments))
94
- return model, transformed
+ return model, transformed
0 commit comments