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 f6d0d18 commit ee8278fCopy full SHA for ee8278f
_unittests/ut_torch_models/test_validate_whole_models2.py
@@ -1,5 +1,6 @@
1
import unittest
2
import onnx
3
+import torch
4
from onnx_diagnostic.ext_test_case import (
5
ExtTestCase,
6
hide_stdout,
@@ -15,6 +16,7 @@ class TestValidateWholeModels2(ExtTestCase):
15
16
@hide_stdout()
17
@ignore_warnings(FutureWarning)
18
@requires_transformers("4.55")
19
+ @unittest.skipif(torch.__version__.startswith("2.9.0"), "no left space space on device?")
20
def test_o_validate_phi35_4k_mini_instruct(self):
21
mid = "microsoft/Phi-3-mini-4k-instruct"
22
summary, data = validate_model(
0 commit comments