Output type for the [`NestedUNet2DConditionModel`] model.
| 3164 | |
| 3165 | |
| 3166 | class NestedUNet2DConditionOutput(BaseOutput): |
| 3167 | """ |
| 3168 | Output type for the [`NestedUNet2DConditionModel`] model. |
| 3169 | """ |
| 3170 | |
| 3171 | sample: list = None |
| 3172 | sample_inner: torch.Tensor = None |
| 3173 | |
| 3174 | |
| 3175 | class NestedUNet2DConditionModel(MatryoshkaUNet2DConditionModel): |
no outgoing calls
no test coverage detected
searching dependent graphs…