outputs.mdx 1.66 KB
Newer Older
Nathan Lambert's avatar
Nathan Lambert committed
1
2
3
4
5
6
7
8
9
10
11
12
<!--Copyright 2022 The HuggingFace Team. All rights reserved.

Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
the License. You may obtain a copy of the License at

http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
specific language governing permissions and limitations under the License.
-->

Patrick von Platen's avatar
Patrick von Platen committed
13
# BaseOutputs
Nathan Lambert's avatar
Nathan Lambert committed
14

Patrick von Platen's avatar
Patrick von Platen committed
15
16
17
All models have outputs that are instances of subclasses of [`~utils.BaseOutput`]. Those are
data structures containing all the information returned by the model, but that can also be used as tuples or
dictionaries.
Nathan Lambert's avatar
Nathan Lambert committed
18

Patrick von Platen's avatar
Patrick von Platen committed
19
Let's see how this looks in an example:
Nathan Lambert's avatar
Nathan Lambert committed
20

Patrick von Platen's avatar
Patrick von Platen committed
21
22
```python
from diffusers import DDIMPipeline
Nathan Lambert's avatar
Nathan Lambert committed
23

Patrick von Platen's avatar
Patrick von Platen committed
24
25
26
pipeline = DDIMPipeline.from_pretrained("google/ddpm-cifar10-32")
outputs = pipeline()
```
Nathan Lambert's avatar
Nathan Lambert committed
27

Patrick von Platen's avatar
Patrick von Platen committed
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
The `outputs` object is a [`~pipeline_utils.ImagePipelineOutput`], as we can see in the
documentation of that class below, it means it has an image attribute.

You can access each attribute as you would usually do, and if that attribute has not been returned by the model, you will get `None`:

```python
outputs.images
```

or via keyword lookup

```python
outputs["images"]
```

When considering our `outputs` object as tuple, it only considers the attributes that don't have `None` values.
Here for instance, we could retrieve images via indexing:

```python
outputs[:1]
```

which will return the tuple `(outputs.images)` for instance.

## BaseOutput

[[autodoc]] utils.BaseOutput
    - to_tuple