Use positional bidirectional mask helper call
#22
by nvidia-oliver-holworthy - opened
Update the custom bidirectional Llama model to call transformers.masking_utils.create_bidirectional_mask with the current inputs_embeds keyword instead of the deprecated input_embeds alias.
nvidia-oliver-holworthy changed pull request title from Use inputs_embeds for bidirectional mask helper to Use positional bidirectional mask helper call
Updated this PR to use positional arguments for create_bidirectional_mask(config, inputs, attention_mask), avoiding dependency on either the deprecated input_embeds keyword or the newer inputs_embeds spelling.
BoLiu changed pull request status to merged