Update modeling_dicow.py

modeling_dicow.py

@@ -274,35 +274,6 @@ class DiCoWForConditionalGeneration(DiCoWGenerationMixin, WhisperForConditionalG
         >>> transcription
         ' Mr. Quilter is the apostle of the middle classes, and we are glad to welcome his gospel.'
         ```"""
-        stno_mask_orig = stno_mask
-        enrollments_processed = None
-        enroll_stno_mask_reshape = None
-        enrollments_enc = None
-        if self.training and self.use_enrollment_network:
-            attention_mask = attention_mask[::2, ...]
-
-            enroll_input = input_features[1::2, ...]
-            input_features = input_features[::2, ...]
-
-            is_valid = is_valid[::2, ...]
-            enroll_stno_mask = stno_mask[1::2, ...]
-            stno_mask = stno_mask[::2, ...]
-
-            labels = labels[::2, ...]
-            upp_labels = upp_labels[::2, ...]
-            enrollments_enc = self.model.encoder.encode_enrollment(
-                input_features=enroll_input,
-                num_layers_to_apply=self.config.spk_embedding_extraction_layer,
-                head_mask=head_mask,
-                stno_mask=enroll_stno_mask,
-            )
-            enroll_stno_mask_reshape = ((enroll_stno_mask[:, 1, :] + enroll_stno_mask[:, 3, :]) > 0.5).view(-1,
-                                                                                                    self.config.mt_num_speakers,
-                                                                                                    enroll_stno_mask.shape[
-                                                                                                        2]).flatten(1,
-                                                                                                                    2)
-            enrollments_processed = enrollments_enc.view(-1, self.config.mt_num_speakers, enrollments_enc.shape[1],
-                                                         enrollments_enc.shape[2]).flatten(1, 2)
 
         return_dict = return_dict if return_dict is not None else self.config.use_return_dict