Skip to content

Commit 07243af

Browse files
authored
fix import bug (#1108)
1 parent 09a0629 commit 07243af

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

neural_compressor/adaptor/onnxrt.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,6 @@
3131
from neural_compressor.utils.utility import Statistics
3232
from neural_compressor.experimental.data.dataloaders.base_dataloader import BaseDataLoader
3333
from neural_compressor.conf.dotdict import deep_get
34-
from neural_compressor.adaptor.ox_utils.util import split_shared_bias
3534
import math
3635

3736
onnx = LazyImport("onnx")
@@ -442,7 +441,8 @@ def _requantize_bias(self, model, bias_name, bias_data):
442441
return new_bias_data
443442

444443
def _pre_optimize(self, model, level=1):
445-
from neural_compressor.adaptor.ox_utils.util import remove_init_from_model_input
444+
from neural_compressor.adaptor.ox_utils.util import \
445+
remove_init_from_model_input, split_shared_bias
446446
remove_init_from_model_input(model)
447447
sess_options = ort.SessionOptions()
448448
level = self.query_handler.get_graph_optimization()

0 commit comments

Comments
 (0)