Revert "feat updated debug pring"
Browse filesThis reverts commit d21ee1bea250c222a68fb85a78d4ef4d356e6e10.
- 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)
|