From 500ef33aa6f657c739f7b693af5be2a830e1c29d Mon Sep 17 00:00:00 2001 From: JingyaHuang Date: Wed, 8 Jan 2025 13:25:21 +0000 Subject: [PATCH] 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)