From f6202aa5998abc9cf7c53d1705c2ff5d66251fff Mon Sep 17 00:00:00 2001 From: JingyaHuang Date: Tue, 7 Jan 2025 14:11:34 +0000 Subject: [PATCH 1/2] fix --- optimum/neuron/modeling_traced.py | 1 + 1 file changed, 1 insertion(+) diff --git a/optimum/neuron/modeling_traced.py b/optimum/neuron/modeling_traced.py index 6d37237c6..ef92da0b0 100644 --- a/optimum/neuron/modeling_traced.py +++ b/optimum/neuron/modeling_traced.py @@ -363,6 +363,7 @@ def _export( local_files_only=local_files_only, token=token, do_validation=False, + library_name=cls.library_name, **kwargs_shapes, ) config = AutoConfig.from_pretrained(save_dir_path) From 500ef33aa6f657c739f7b693af5be2a830e1c29d Mon Sep 17 00:00:00 2001 From: JingyaHuang Date: Wed, 8 Jan 2025 13:25:21 +0000 Subject: [PATCH 2/2] fix test --- tests/inference/test_modeling.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tests/inference/test_modeling.py b/tests/inference/test_modeling.py index 51c873db0..dd6d3ffc7 100644 --- a/tests/inference/test_modeling.py +++ b/tests/inference/test_modeling.py @@ -109,7 +109,6 @@ def test_load_model_from_hub_subfolder(self): self.TINY_SUBFOLDER_MODEL_ID, subfolder="my_subfolder", export=True, - library_name="transformers", **self.STATIC_INPUTS_SHAPES, ) self.assertIsInstance(model.model, torch.jit._script.ScriptModule)