Apppublic

declare-lab/tango2

sourceHugging Faceupdated 2y agoView on Hugging Face

92likes

rootdiffusersteststest_pipelines_onnx_common.py

test_pipelines_onnx_common.py13 linesDownload Raw Back to tests

1from diffusers.utils.testing_utils import require_onnxruntime2 3 4@require_onnxruntime5class 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    pass13