Skip to content

Commit f95e299

Browse files
committed
remove one unit test
1 parent a9b2581 commit f95e299

File tree

1 file changed

+0
-10
lines changed

1 file changed

+0
-10
lines changed

_unittests/ut_torch_models/test_hghub_model.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -75,16 +75,6 @@ def test_get_untrained_model_with_inputs_beit(self):
7575
# different expected value for different version of transformers
7676
self.assertIn((data["size"], data["n_weights"]), [(111448, 27862), (56880, 14220)])
7777

78-
@hide_stdout()
79-
@ignore_errors(OSError)
80-
def test_get_untrained_model_with_inputs_codellama(self):
81-
mid = "codellama/CodeLlama-7b-Python-hf"
82-
data = get_untrained_model_with_inputs(mid, verbose=1)
83-
model, inputs = data["model"], data["inputs"]
84-
model(**inputs)
85-
# different expected value for different version of transformers
86-
self.assertIn((data["size"], data["n_weights"]), [(547377152, 136844288)])
87-
8878
@hide_stdout()
8979
@ignore_errors(OSError)
9080
def test_get_untrained_model_with_inputs_clip_vit(self):

0 commit comments

Comments
 (0)