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]>
- 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.
|
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
|