"official/vision/beta/modeling/classification_model.py" did not exist on "2f788e1dcce7bc7982ecfa3ba03b26429a6303b0"
test_pipelines_onnx_common.py 367 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
from diffusers.utils.testing_utils import require_onnxruntime


@require_onnxruntime
class OnnxPipelineTesterMixin:
    """
    This mixin is designed to be used with unittest.TestCase classes.
    It provides a set of common tests for each ONNXRuntime pipeline, e.g. saving and loading the pipeline,
    equivalence of dict and tuple outputs, etc.
    """

    pass