Commit cf062905 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

remove ipdb

parent 374deef4
......@@ -641,9 +641,6 @@ class TFPreTrainedModel(tf.keras.Model, TFModelUtilsMixin):
# create attention mask if necessary
# TODO (PVP): this should later be handled by the forward fn() in each model in the future see PR 3140
import ipdb
ipdb.set_trace()
if (attention_mask is None) and (pad_token_id is not None) and (pad_token_id in input_ids.numpy()):
attention_mask = tf.cast(tf.math.not_equal(input_ids, pad_token_id), dtype=tf.int32)
elif attention_mask is None:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment