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

Add 'add_extra_input' to handle models like QLora #370

Merged
merged 5 commits into from
May 1, 2024

Conversation

RyanUnderhill
Copy link
Member

Add a new python api 'add_extra_input' that will take numpy tensors and turn them into OrtValue inputs internally.
This allows models with extra custom inputs (like QLora) to be specified in python.

C API to follow soon.

@RyanUnderhill RyanUnderhill merged commit b3ff5ce into main May 1, 2024
12 checks passed
@RyanUnderhill RyanUnderhill deleted the ryanunderhill/custom_inputs branch May 1, 2024 04:38
baijumeswani pushed a commit that referenced this pull request May 10, 2024
Add a new python api 'add_extra_input' that will take numpy tensors and
turn them into OrtValue inputs internally.
This allows models with extra custom inputs (like QLora) to be specified
in python.

C API to follow soon.
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.

3 participants