Markus28 commited on
Commit
a62c2ab
·
1 Parent(s): d21ee1b

Revert "feat updated debug pring"

Browse files

This reverts commit d21ee1bea250c222a68fb85a78d4ef4d356e6e10.

Files changed (1) hide show
  1. modeling_bert.py +0 -1
modeling_bert.py CHANGED
@@ -424,7 +424,6 @@ class BertModel(BertPreTrainedModel):
424
 
425
  def to(self, target):
426
  result = super().to(target)
427
- print(f'Calling .to({target})')
428
  if isinstance(target, torch.dtype):
429
  for layer in result.encoder.layers:
430
  layer.mixer.inner_cross_attn.alibi_slopes = layer.mixer.inner_cross_attn.alibi_slopes.to(torch.float32)
 
424
 
425
  def to(self, target):
426
  result = super().to(target)
 
427
  if isinstance(target, torch.dtype):
428
  for layer in result.encoder.layers:
429
  layer.mixer.inner_cross_attn.alibi_slopes = layer.mixer.inner_cross_attn.alibi_slopes.to(torch.float32)