Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
shivamshrirao
GitHub Repository: shivamshrirao/diffusers
Path: blob/main/tests/test_pipelines_onnx_common.py
1441 views
1
from diffusers.utils.testing_utils import require_onnxruntime
2
3
4
@require_onnxruntime
5
class OnnxPipelineTesterMixin:
6
"""
7
This mixin is designed to be used with unittest.TestCase classes.
8
It provides a set of common tests for each ONNXRuntime pipeline, e.g. saving and loading the pipeline,
9
equivalence of dict and tuple outputs, etc.
10
"""
11
12
pass
13
14