Fixed one more issue

huggingface · Oct 3, 2023 · 9adc298 · 9adc298
1 parent aa0c6ad
commit 9adc298
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/optimum/exporters/openvino/__main__.py b/optimum/exporters/openvino/__main__.py
@@ -234,11 +234,10 @@ def main_export(
         onnx_config_constructor = TasksManager.get_exporter_config_constructor(model=model, exporter="onnx", task=task)
         onnx_config = onnx_config_constructor(model.config)
         models_and_onnx_configs = {"model": (model, onnx_config)}
+        if model_kwargs is None:
+            model_kwargs = {}
         load_in_8bit = model_kwargs.get("load_in_8bit", None)
         if load_in_8bit is None:
-            if model_kwargs is None:
-                model_kwargs = {}
-
             if model.num_parameters() >= _MAX_UNCOMPRESSED_DECODER_SIZE:
                 model_kwargs["load_in_8bit"] = True
             else: