class_labels in UNet3DConditionModel

[KAI10]

Hi,

I am going through the source code of UNet3DConditionModel, specifically the forward method. Here a parameter named class_labels is available with the description:

class_labels (torch.Tensor, optional, defaults to None):
Optional class labels for conditioning. Their embeddings will be summed with the timestep embeddings.

However, it seems this parameter is not used in the method. Is this part not implemented yet?

Thanks in advance!

[asomoza]

cc @sayakpaul

[sayakpaul]

Yes, class_labels isn't implemented yet for the 3D variant.