File tree Expand file tree Collapse file tree 3 files changed +4
-6
lines changed
neural_compressor/torch/algorithms/fp8_quant Expand file tree Collapse file tree 3 files changed +4
-6
lines changed Original file line number Diff line number Diff line change @@ -13,10 +13,10 @@ def update_mod_dict(config):
13
13
14
14
def print_init_info (config ):
15
15
import importlib .metadata
16
- versionStr = importlib .metadata .version ('habana_quantization_toolkit ' )
16
+ versionStr = importlib .metadata .version ('neural_compressor_3x_pt ' )
17
17
locationStr = versionStr .find ('git' ) + 3
18
- logger .info ("HQT Git revision = %s" , versionStr [locationStr :])
19
- logger .info ("HQT Configuration = %s" , config )
18
+ logger .info ("neural_compressor_3x_pt Git revision = %s" , versionStr [locationStr :])
19
+ logger .info ("neural_compressor_3x_pt Configuration = %s" , config )
20
20
21
21
def is_substr (substr_list , target ):
22
22
return any ([x in target for x in substr_list ])
Original file line number Diff line number Diff line change @@ -70,7 +70,7 @@ def create_mod_info_recursion(parent):
70
70
71
71
72
72
def get_patched_mod_list ():
73
- from habana_quantization_toolkit ._core .common import mod_default_dict
73
+ from ._core .common import mod_default_dict
74
74
75
75
patched_mod_list = []
76
76
for patched_mod in mod_default_dict .values ():
Original file line number Diff line number Diff line change @@ -54,8 +54,6 @@ def _convert(model, config_path):
54
54
55
55
56
56
def _prepare (model , config_path ):
57
- import habana_quantization_toolkit as hqt
58
-
59
57
# update mode to MEASURE
60
58
config_path = update_mode (config_path , measure_step = True )
61
59
You can’t perform that action at this time.
0 commit comments