We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 28dbb68 commit 8e1fc8bCopy full SHA for 8e1fc8b
1 file changed
_unittests/ut_torch_models/test_tiny_llms_bypassed.py
@@ -1,7 +1,12 @@
1
import copy
2
import unittest
3
import torch
4
-from onnx_diagnostic.ext_test_case import ExtTestCase, ignore_warnings, hide_stdout, requires_transformers
+from onnx_diagnostic.ext_test_case import (
5
+ ExtTestCase,
6
+ ignore_warnings,
7
+ hide_stdout,
8
+ requires_transformers,
9
+)
10
from onnx_diagnostic.torch_models.llms import get_tiny_llm
11
from onnx_diagnostic.torch_models.llms import get_phi2
12
from onnx_diagnostic.helpers import string_type
0 commit comments