Skip to content

[llava][10/N] Fix use_sdpa_with_kv_cache option #4456

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

Closed

Conversation

larryliu0820
Copy link
Contributor

@larryliu0820 larryliu0820 commented Jul 30, 2024

Stack from ghstack (oldest at bottom):

Summary: As titled. In export_llava.py export_text_model() needs to
respect use_sdpa_with_kv_cache_op option.

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:

Differential Revision: D60431561

Summary: As titled. In `export_llava.py` `export_text_model()` needs to
respect `use_sdpa_with_kv_cache_op` option.

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:

[ghstack-poisoned]
Copy link

pytorch-bot bot commented Jul 30, 2024

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/4456

Note: Links to docs will display an error until the docs builds have been completed.

✅ No Failures

As of commit 69c5b10 with merge base 1e14333 (image):
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.

larryliu0820 added a commit that referenced this pull request Jul 30, 2024
Summary: As titled. In `export_llava.py` `export_text_model()` needs to
respect `use_sdpa_with_kv_cache_op` option.

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:

ghstack-source-id: 4e6387c
Pull Request resolved: #4456
@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Jul 30, 2024
@larryliu0820
Copy link
Contributor Author

@larryliu0820 has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

@larryliu0820 larryliu0820 requested a review from tarun292 July 30, 2024 03:01
@facebook-github-bot
Copy link
Contributor

This pull request has been merged in 28cfabb.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. Merged
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants