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)