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

Allow Passing Model Keyword Arguments Through the parallelize Interface #97

Merged
merged 4 commits into from
Oct 25, 2024

Conversation

Dando18
Copy link
Contributor

@Dando18 Dando18 commented Sep 17, 2024

Allow passing model keyword arguments through parallelize to AutoConfig.from_pretrained.

This primarily came about because AutoConfig.from_pretrained needs to be passed trust_remote_code=True when a model repo has custom code, otherwise the config loading will fail. This will simply allow passing any custom params to the AutoConfig interface.

@siddharth9820
Copy link
Collaborator

siddharth9820 commented Sep 18, 2024

lgtm. Will run the CI and merge this. Thank you

@siddharth9820 siddharth9820 self-requested a review October 25, 2024 15:16
@siddharth9820 siddharth9820 merged commit 6252013 into develop Oct 25, 2024
4 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request ready-for-review
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants