Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix llama_v2_7b_16h for torch.jit.trace #2121

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

thiagocrepaldi
Copy link
Contributor

Original error: Attention using SDPA can not be traced with torch.jit.trace when no attention_mask is provided. To solve this issue, please either load your model with the argument attn_implementation="eager" or pass an attention_mask input when tracing the model.

Original error: Attention using SDPA can not be traced with torch.jit.trace
when no attention_mask is provided. To solve this issue, please either load
your model with the argument attn_implementation="eager" or
pass an attention_mask input when tracing the model.
@thiagocrepaldi
Copy link
Contributor Author

@huydhn do you think that change addresses #117752?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants