It seems that the number of input features in self.adv_layer or self.aux_layer is defined as 512, while the incoming activation has 768 features.
You could fix this error by setting in_features=768 for the linear layers.