Skip to content

Commit a505dc6

Browse files
Reformated onnx_transform file
1 parent de34841 commit a505dc6

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
@@ -6,11 +6,11 @@
66
# ----------------------------------------------------------------------------
77

88
import os
9-
import numpy as np
10-
from typing import Optional, Tuple
119
from concurrent.futures import ThreadPoolExecutor
12-
from onnx import ModelProto, external_data_helper, numpy_helper
10+
from typing import Optional, Tuple
1311

12+
import numpy as np
13+
from onnx import ModelProto, external_data_helper, numpy_helper
1414

1515

1616
class OnnxTransform:
@@ -91,4 +91,4 @@ def process_tensor(args):
9191

9292
with ThreadPoolExecutor(max_workers=os.cpu_count() * 4) as executor:
9393
list(executor.map(process_tensor, file_assignments))
94-
return model, transformed
94+
return model, transformed

0 commit comments

Comments
 (0)