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

Supporting AutoModelForCausalLM #385

Closed
jis478 opened this issue Apr 23, 2024 · 4 comments
Closed

Supporting AutoModelForCausalLM #385

jis478 opened this issue Apr 23, 2024 · 4 comments

Comments

@jis478
Copy link

jis478 commented Apr 23, 2024

Hi, I'm writing a new class like this for models in the AutoModelForCausalLM format. Originally I've been testing sql-coder model with this but thought this could be generalized as a class. Do you have an on-going PR or can i make one?

@zainhoda
Copy link
Contributor

This isn't something that I've heard from other users so please feel free to make a PR!

@jis478
Copy link
Author

jis478 commented Apr 29, 2024

Thanks, I've made PR (#391)

@jis478
Copy link
Author

jis478 commented Apr 30, 2024

By the way the original Sql-coder repo is also implemented using the Huggingface Transformers library with its own prompt template. As pointed out in #303 (comment), it would be great if you tested it out again with the original repo (I'm currently working on getting it to run on Vanna, based on this PR as well)

@zainhoda zainhoda closed this as completed May 2, 2024
@andreped
Copy link
Contributor

andreped commented May 2, 2024

By the way the original Sql-coder repo is also implemented using the Hugging Face Transformers library with its own prompt template. As pointed out in #303 (comment), it would be great if you tested it out again (I'm currently working on getting it to run on Vanna, based on this PR as well)

Would be of interest to me as well, @zainhoda. Great suggestion, @jis478! :]

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

No branches or pull requests

3 participants