Skip to content

Commit

Permalink
Remove mask if use fusion mask (NVIDIA#9723)
Browse files Browse the repository at this point in the history
* Remove mask if use fusion mask

Signed-off-by: Cheng-Ping Hsieh <chsieh@nvidia.com>

* Apply isort and black reformatting

Signed-off-by: hsiehjackson <hsiehjackson@users.noreply.github.com>

---------

Signed-off-by: Cheng-Ping Hsieh <chsieh@nvidia.com>
Signed-off-by: hsiehjackson <hsiehjackson@users.noreply.github.com>
Co-authored-by: hsiehjackson <hsiehjackson@users.noreply.github.com>
Signed-off-by: Malay Nagda <malayn@malayn-mlt.client.nvidia.com>
  • Loading branch information
2 people authored and Malay Nagda committed Jul 26, 2024
1 parent 4667136 commit aa3dfa0
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions nemo/collections/nlp/modules/common/text_generation_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -794,6 +794,9 @@ def generate(
if random_seed is not None:
seed_everything(random_seed)

if hasattr(model, 'get_attention_mask_from_fusion') and model.get_attention_mask_from_fusion:
compute_attention_mask = False

output = synced_generate(
model,
inference_strategy,
Expand Down

0 comments on commit aa3dfa0

Please sign in to comment.