Skip to content

输出

所有模型输出都是 [~utils.BaseOutput] 的子类,包含模型返回的所有信息的数据结构。输出也可以用作元组或字典。

例如:

python
from diffusers import DDIMPipeline

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

outputs 对象是一个 [~pipelines.ImagePipelineOutput],这意味着它有一个 image 属性。

你可以像平常一样访问每个属性,或者使用关键字查找,如果模型没有返回该属性,你将得到 None

python
outputs.images
outputs["images"]

在将 outputs 对象视为元组时,它只考虑那些不为 None 的属性。 例如,通过索引获取图像会返回元组 (outputs.images)

python
outputs[:1]

BaseOutput

[[autodoc]] utils.BaseOutput - to_tuple

ImagePipelineOutput

[[autodoc]] pipelines.ImagePipelineOutput

FlaxImagePipelineOutput

[[autodoc]] pipelines.pipeline_flax_utils.FlaxImagePipelineOutput

AudioPipelineOutput

[[autodoc]] pipelines.AudioPipelineOutput

ImageTextPipelineOutput

[[autodoc]] ImageTextPipelineOutput