mrprimenotes commited on
Commit
2205c69
·
verified ·
1 Parent(s): 53c2202

Remove Debugging

Browse files
Files changed (1) hide show
  1. model.py +0 -2
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
  )