ShadeEngine's picture
End of training
4ac8f3e verified

Outputs

All model outputs are subclasses of [~utils.BaseOutput], data structures containing all the information returned by the model. The outputs can also be used as tuples or dictionaries.

For example:

from diffusers import DDIMPipeline

pipeline = DDIMPipeline.from_pretrained("google/ddpm-cifar10-32")
outputs = pipeline()

The outputs object is a [~pipelines.ImagePipelineOutput] which means it has an image attribute.

You can access each attribute as you normally would or with a keyword lookup, and if that attribute is not returned by the model, you will get None:

outputs.images
outputs["images"]

When considering the outputs object as a tuple, it only considers the attributes that don't have None values. For instance, retrieving an image by indexing into it returns the tuple (outputs.images):

outputs[:1]

To check a specific pipeline or model output, refer to its corresponding API documentation.

BaseOutput

[[autodoc]] utils.BaseOutput - to_tuple

ImagePipelineOutput

[[autodoc]] pipelines.ImagePipelineOutput

FlaxImagePipelineOutput

[[autodoc]] pipelines.pipeline_flax_utils.FlaxImagePipelineOutput

AudioPipelineOutput

[[autodoc]] pipelines.AudioPipelineOutput

ImageTextPipelineOutput

[[autodoc]] ImageTextPipelineOutput