We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 74a4130 commit ca20fb5Copy full SHA for ca20fb5
_unittests/ut_torch_models/test_tiny_llms.py
@@ -48,6 +48,7 @@ def test_tiny_llm_run_static(self):
48
49
@ignore_warnings(UserWarning)
50
@requires_transformers("4.52")
51
+ @requires_torch("2.8")
52
def test_tiny_llm_export_static(self):
53
data = get_tiny_llm(use_static_cache=True)
54
model, inputs = data["model"], data["inputs"]
0 commit comments