File tree 1 file changed +5
-1
lines changed
1 file changed +5
-1
lines changed Original file line number Diff line number Diff line change @@ -103,14 +103,18 @@ def __init__(
103
103
def load_model (
104
104
file_name : Union [str , Path ],
105
105
quantization_config : Union [OVWeightQuantizationConfig , Dict ] = None ,
106
- calibration_dataset : Optional [ "Dataset" ] = None ,
106
+ calibration_dataset : Optional = None ,
107
107
):
108
108
"""
109
109
Loads the model.
110
110
111
111
Arguments:
112
112
file_name (`str` or `Path`):
113
113
The path of the model ONNX or XML file.
114
+ quantization_config (`OVWeightQuantizationConfig` or `Dict`, *optional*):
115
+ Quantization config to apply after model is loaded.
116
+ calibration_dataset (`nncf.Dataset`, *optional*):
117
+ Optional nncf.Dataset to feed to model weight compression when quantization config is provided.
114
118
"""
115
119
116
120
def fix_op_names_duplicates (model : openvino .runtime .Model ):
You can’t perform that action at this time.
0 commit comments