Diffusers documentation
Salidas
Salidas
Todas las salidas de los modelos son subclases de BaseOutput, estructuras de datos que contienen toda la información devuelta por el modelo. Las salidas también se pueden usar como tuplas o diccionarios.
Por ejemplo:
from diffusers import DDIMPipeline
pipeline = DDIMPipeline.from_pretrained("google/ddpm-cifar10-32")
outputs = pipeline()El objeto outputs es un ImagePipelineOutput, lo que significa que tiene un atributo de imagen.
Puedes acceder a cada atributo de la forma habitual o mediante una búsqueda por palabra clave, y si ese atributo no es devuelto por el modelo, obtendrás None:
outputs.images
outputs["images"]Al considerar el objeto outputs como una tupla, solo se tienen en cuenta los atributos que no tienen valores None.
Por ejemplo, recuperar una imagen mediante su índice devuelve la tupla (outputs.images):
outputs[:1]Para verificar la salida de un pipeline o modelo específico, consulte la documentación de la API correspondiente.
BaseOutput
Clase base para todas las salidas de modelos como dataclass. Tiene un __getitem__ que permite indexar por entero o por rebanada (como una tupla) o por cadenas (como un diccionario), ignorando los atributos None. De lo contrario, se comporta como un diccionario de Python regular.
> No se puede desempaquetar un
BaseOutputdirectamente. Usa el método to_tuple() para convertirlo en una tupla > primero.
Convierte self en una tupla que contiene todos los atributos/llaves que no son None.
ImagePipelineOutput
clase diffusers.ImagePipelineOutput
< source >( images: list[PIL.Image.Image] | numpy.ndarray )
Clase de salida para pipelines de imágenes.
AudioPipelineOutput
clase diffusers.AudioPipelineOutput
< source >( audios: ndarray )
Clase de salida para las tuberías de audio.
ImageTextPipelineOutput
clase diffusers.ImageTextPipelineOutput
< source >( images: list[PIL.Image.Image] | numpy.ndarray | Nonetext: list[str] | list[list[str]] | None )
Parámetros
- images (
list[PIL.Image.Image]onp.ndarray) — lista de imágenes PIL desruidadas de longitudbatch_sizeo array de NumPy de forma(batch_size, height, width, num_channels). - text (
list[str]olist[list[str]]) — lista de cadenas de texto generadas de longitudbatch_sizeo una lista de listas de cadenas cuya lista externa tiene longitudbatch_size.
Clase de salida para pipelines conjuntos de imagen y texto.