Diffusers documentation
Output
Output
Semua keluaran model adalah subkelas dari BaseOutput, struktur data yang berisi semua informasi yang dikembalikan oleh model. Keluaran tersebut juga dapat digunakan sebagai tuple atau kamus.
Sebagai contoh:
from diffusers import DDIMPipeline
pipeline = DDIMPipeline.from_pretrained("google/ddpm-cifar10-32")
outputs = pipeline()Objek outputs adalah ImagePipelineOutput, yang berarti memiliki atribut image.
Anda dapat mengakses setiap atribut seperti biasa atau dengan pencarian kata kunci, dan jika atribut tersebut tidak dikembalikan oleh model, Anda akan mendapatkan None:
outputs.images
outputs["images"]Saat objek outputs dipertimbangkan sebagai tuple, hanya atribut yang tidak memiliki nilai None yang dipertimbangkan.
Sebagai contoh, mengambil gambar dengan mengindeksnya mengembalikan tuple (outputs.images):
outputs[:1]Untuk memeriksa output pipeline atau model tertentu, lihat dokumentasi API yang sesuai.
BaseOutput
Kelas dasar untuk semua output model sebagai dataclass. Memiliki __getitem__ yang memungkinkan indeksasi dengan integer atau slice (seperti tuple) atau string (seperti dictionary) yang akan mengabaikan atribut None. Selain itu, berperilaku seperti dictionary Python biasa.
> Anda tidak dapat mengurai
BaseOutputsecara langsung. Gunakan metode to_tuple() untuk mengubahnya menjadi tuple > terlebih dahulu.
Konversi self menjadi tuple yang berisi semua atribut/kunci yang bukan None.
ImagePipelineOutput
kelas diffusers.ImagePipelineOutput
< source >( images: list[PIL.Image.Image] | numpy.ndarray )
Kelas output untuk pipeline gambar.
AudioPipelineOutput
kelas diffusers.AudioPipelineOutput
< source >( audios: ndarray )
Kelas output untuk pipeline audio.
ImageTextPipelineOutput
kelas diffusers.ImageTextPipelineOutput
< source >( images: list[PIL.Image.Image] | numpy.ndarray | Nonetext: list[str] | list[list[str]] | None )
Parameter
- images (
list[PIL.Image.Image]ataunp.ndarray) — daftar gambar PIL yang telah didenoise dengan panjangbatch_sizeatau array NumPy dengan bentuk(batch_size, height, width, num_channels). - text (
list[str]ataulist[list[str]]) — daftar string teks yang dihasilkan dengan panjangbatch_sizeatau daftar dari daftar string di mana daftar luarnya memiliki panjangbatch_size.
Kelas output untuk pipeline gambar-teks gabungan.