Make UNet2DConditionOutput
pickle-able
#3857
Merged
UNet2DConditionOutput
pickle-able
#3857