"git@developer.sourcefind.cn:zhaoyu6/sglang.git" did not exist on "852c0578fd3377a980cf2296751424a56052155e"
test_pipelines_common.py 347 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
from diffusers.utils.testing_utils import require_torch


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

    pass