From 715c054360946b312e94beec91c55a1223258954 Mon Sep 17 00:00:00 2001 From: Ekaterina Aidova Date: Fri, 17 May 2024 13:38:58 +0400 Subject: [PATCH] Skip saving gen config if saving failed (#717) --- optimum/exporters/openvino/convert.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/optimum/exporters/openvino/convert.py b/optimum/exporters/openvino/convert.py index baa34a5cd0..3b214f77e4 100644 --- a/optimum/exporters/openvino/convert.py +++ b/optimum/exporters/openvino/convert.py @@ -614,7 +614,12 @@ def export_from_model( model.config.save_pretrained(output) generation_config = getattr(model, "generation_config", None) if generation_config is not None: - generation_config.save_pretrained(output) + try: + generation_config.save_pretrained(output) + except Exception as exception: + logger.warning( + f"The generation config will not be saved, saving failed with following error:\n{exception}" + ) model_name_or_path = model.config._name_or_path maybe_save_preprocessors(model_name_or_path, output, trust_remote_code=trust_remote_code)