Skip to content

Commit 70ee0ef

Browse files
Fix ruff 2
1 parent f314ba0 commit 70ee0ef

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

optimum/intel/openvino/modeling_base.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -103,14 +103,18 @@ def __init__(
103103
def load_model(
104104
file_name: Union[str, Path],
105105
quantization_config: Union[OVWeightQuantizationConfig, Dict] = None,
106-
calibration_dataset: Optional["Dataset"] = None,
106+
calibration_dataset: Optional = None,
107107
):
108108
"""
109109
Loads the model.
110110
111111
Arguments:
112112
file_name (`str` or `Path`):
113113
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.
114118
"""
115119

116120
def fix_op_names_duplicates(model: openvino.runtime.Model):

0 commit comments

Comments
 (0)