Spaces:
Sleeping
Sleeping
File size: 481 Bytes
1bf3c29 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | from dataclasses import dataclass
from typing import List, Union
import numpy as np
import PIL.Image
from diffusers.utils import BaseOutput
@dataclass
class PixelDiTPipelineOutput(BaseOutput):
"""
Output class for PixelDiT text-to-image pipelines.
Args:
images (`List[PIL.Image.Image]` or `np.ndarray`):
List of generated images or numpy array of shape `(batch, height, width, 3)`.
"""
images: Union[List[PIL.Image.Image], np.ndarray]
|