Spaces:

madtune
/

pixeldit-space

Running

pixeldit-space / diffusers_patch /src /diffusers /pipelines /pixeldit /pipeline_output.py

add pipeline patch and scripts

c85ad6e 5 months ago

481 Bytes

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]

	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]