SinclairSchneider winglian commited on
Commit
f5274db
1 Parent(s): 4bee1a6

reduce verbosity of logging (#1)

Browse files

- reduce verbosity of logging (3f5546e29a9afce28e2b42ecce43b34dacedfbc2)


Co-authored-by: wing lian <[email protected]>

Files changed (1) hide show
  1. modeling_dbrx.py +1 -1
modeling_dbrx.py CHANGED
@@ -411,7 +411,7 @@ class DbrxFlashAttention2(DbrxAttention):
411
  **kwargs: Any,
412
  ) -> Tuple[torch.Tensor, Optional[torch.Tensor],
413
  Optional[Tuple[torch.Tensor]]]:
414
- logger.info(
415
  'Implicitly setting `output_attentions` to False as it is not supported in Flash Attention.'
416
  )
417
  output_attentions = False
 
411
  **kwargs: Any,
412
  ) -> Tuple[torch.Tensor, Optional[torch.Tensor],
413
  Optional[Tuple[torch.Tensor]]]:
414
+ logger.debug(
415
  'Implicitly setting `output_attentions` to False as it is not supported in Flash Attention.'
416
  )
417
  output_attentions = False