File tree 1 file changed +5
-7
lines changed
1 file changed +5
-7
lines changed Original file line number Diff line number Diff line change @@ -118,14 +118,12 @@ def __init__(
118
118
self .save_onnx_model = save_onnx_model
119
119
self ._enable_standard_onnx_export_option ()
120
120
self .optimum_version = kwargs .pop ("optimum_version" , None )
121
- self .quantization_config = quantization_config
122
-
123
- bits = None
124
- if isinstance (quantization_config , dict ):
125
- bits = quantization_config .get ("bits" , None )
126
- elif isinstance (quantization_config , QuantizationConfigMixin ):
127
- bits = quantization_config .bits
121
+ self .quantization_config = quantization_config or {}
128
122
123
+ if isinstance (quantization_config , QuantizationConfigMixin ):
124
+ bits = self .quantization_config .bits
125
+ else :
126
+ bits = self .quantization_config .get ("bits" , None )
129
127
self .dtype = "int" + str (bits ) if isinstance (bits , int ) else dtype
130
128
131
129
def add_input_info (self , model_inputs : Dict , force_batch_one : bool = False ):
You can’t perform that action at this time.
0 commit comments