Remove Debugging
Browse files
model.py
CHANGED
@@ -2268,14 +2268,12 @@ class CustomWhisperForConditionalGeneration(CustomWhisperGenerationMixin, Whispe
|
|
2268 |
```"""
|
2269 |
return_dict = return_dict if return_dict is not None else self.config.use_return_dict
|
2270 |
|
2271 |
-
print(labels)
|
2272 |
if labels is not None:
|
2273 |
if labels.shape[1] > self.max_target_positions:
|
2274 |
raise ValueError(
|
2275 |
f"Labels' sequence length {labels.shape[1]} cannot exceed the maximum allowed length of {self.max_target_positions} tokens."
|
2276 |
)
|
2277 |
if decoder_input_ids is None and decoder_inputs_embeds is None:
|
2278 |
-
print(f"shift_tokens_right")
|
2279 |
decoder_input_ids = shift_tokens_right(
|
2280 |
labels, self.config.pad_token_id, self.config.decoder_start_token_id
|
2281 |
)
|
|
|
2268 |
```"""
|
2269 |
return_dict = return_dict if return_dict is not None else self.config.use_return_dict
|
2270 |
|
|
|
2271 |
if labels is not None:
|
2272 |
if labels.shape[1] > self.max_target_positions:
|
2273 |
raise ValueError(
|
2274 |
f"Labels' sequence length {labels.shape[1]} cannot exceed the maximum allowed length of {self.max_target_positions} tokens."
|
2275 |
)
|
2276 |
if decoder_input_ids is None and decoder_inputs_embeds is None:
|
|
|
2277 |
decoder_input_ids = shift_tokens_right(
|
2278 |
labels, self.config.pad_token_id, self.config.decoder_start_token_id
|
2279 |
)
|