Skip to content

Commit

Permalink
handle empty quant dictionary
Browse files Browse the repository at this point in the history
  • Loading branch information
IlyasMoutawwakil committed May 28, 2024
1 parent 333b7b0 commit d32f910
Showing 1 changed file with 8 additions and 4 deletions.
12 changes: 8 additions & 4 deletions optimum/onnxruntime/configuration.py
Original file line number Diff line number Diff line change
Expand Up @@ -994,18 +994,22 @@ def __init__(
self.use_external_data_format = use_external_data_format
self.one_external_file = one_external_file

if isinstance(optimization, dict):
if isinstance(optimization, dict) and optimization:
self.optimization = OptimizationConfig(**optimization)
elif isinstance(optimization, OptimizationConfig) or optimization is None:
elif isinstance(optimization, OptimizationConfig):
self.optimization = optimization
elif not optimization:
self.optimization = None
else:
raise ValueError(
f"Optional argument `optimization` must be a dictionary or an instance of OptimizationConfig, got {type(optimization)}"
)
if isinstance(quantization, dict):
if isinstance(quantization, dict) and quantization:
self.quantization = QuantizationConfig(**quantization)
elif isinstance(quantization, QuantizationConfig) or quantization is None:
elif isinstance(quantization, QuantizationConfig):
self.quantization = quantization
elif not quantization:
self.quantization = None
else:
raise ValueError(
f"Optional argument `quantization` must be a dictionary or an instance of QuantizationConfig, got {type(quantization)}"
Expand Down

0 comments on commit d32f910

Please sign in to comment.