Skip to content

Commit

Permalink
oops
Browse files Browse the repository at this point in the history
  • Loading branch information
younesbelkada committed Mar 4, 2024
1 parent adc3891 commit 7377d95
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 3 deletions.
4 changes: 1 addition & 3 deletions .github/workflows/push-important-models.yml
Original file line number Diff line number Diff line change
Expand Up @@ -74,7 +74,6 @@ jobs:
slack_channel: ${{ env.OUTPUT_SLACK_CHANNEL_ID }}
title: FA2 tests - ${{ matrix.model-name }}
status: ${{ steps.run_fa2_tests.conclusion}}
secrets: inherit

- name: Run integration tests
id: run_integration_tests
Expand All @@ -87,5 +86,4 @@ jobs:
with:
slack_channel: ${{ env.OUTPUT_SLACK_CHANNEL_ID }}
title: Integration tests - ${{ matrix.model-name }}
status: ${{ steps.run_integration_tests.conclusion}}
secrets: inherit
status: ${{ steps.run_integration_tests.conclusion}}
1 change: 1 addition & 0 deletions src/transformers/models/llama/modeling_llama.py
Original file line number Diff line number Diff line change
Expand Up @@ -59,6 +59,7 @@

_CONFIG_FOR_DOC = "LlamaConfig"


def _get_unpad_data(attention_mask):
seqlens_in_batch = attention_mask.sum(dim=-1, dtype=torch.int32)
indices = torch.nonzero(attention_mask.flatten(), as_tuple=False).flatten()
Expand Down

0 comments on commit 7377d95

Please sign in to comment.