Skip to content

Commit 8e1fc8b

Browse files
committed
fix
1 parent 28dbb68 commit 8e1fc8b

1 file changed

Lines changed: 6 additions & 1 deletion

File tree

_unittests/ut_torch_models/test_tiny_llms_bypassed.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,12 @@
11
import copy
22
import unittest
33
import torch
4-
from onnx_diagnostic.ext_test_case import ExtTestCase, ignore_warnings, hide_stdout, requires_transformers
4+
from onnx_diagnostic.ext_test_case import (
5+
ExtTestCase,
6+
ignore_warnings,
7+
hide_stdout,
8+
requires_transformers,
9+
)
510
from onnx_diagnostic.torch_models.llms import get_tiny_llm
611
from onnx_diagnostic.torch_models.llms import get_phi2
712
from onnx_diagnostic.helpers import string_type

0 commit comments

Comments
 (0)