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

[LLM Tests] Config support for LLM test suite / enable test suite in GHA #1324

Merged
merged 15 commits into from
Oct 23, 2023

Conversation

dbogunowicz
Copy link
Contributor

@dbogunowicz dbogunowicz commented Oct 17, 2023

Configurable testing suite for LLMs.

Introduces two types of tests:

  1. Unit tests - test the smaller functionalities of the LLM pipelines (text_generation and chat). Those are small, self-contained lightweight tests that use the tinystories model. Those will run on every commit in GHA. It should be easy and simple to add new test cases here by any developer.

  2. Integration test for LLMs - test end-to-end all the relevant LLM pipelines. Those are integration tests that are configurable and by default will not run on every commit but with some lower frequency. So far we have tests for codegen and opt models. I have run a limited scope of these tests on my local machine, so I do not guarantee that they will instantaneously all pass on the first go once deployed. Codegen has been thoroughly tested by me, but opt, because of its size, not.

Along with this feature come small fixes to the LLM pipelines that were required to make tests work properly.

@dbogunowicz dbogunowicz changed the title [WiP][LLM Tests] Config support for LLM test suite / enable test suite in GHA [LLM Tests] Config support for LLM test suite / enable test suite in GHA Oct 17, 2023
@dbogunowicz
Copy link
Contributor Author

Tests are green yay! :)

Copy link
Contributor

@Satrat Satrat left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Nice work on this!

@dbogunowicz dbogunowicz merged commit 67327cf into main Oct 23, 2023
11 of 13 checks passed
@dbogunowicz dbogunowicz deleted the feature/damian/final_llm_tests branch October 23, 2023 20:40
dbogunowicz added a commit that referenced this pull request Oct 25, 2023
Reverting my change from #1324 
Back then, I was not aware that this property of the sparsezoo `Model` was an intended change (assumed it was a typo)
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

Successfully merging this pull request may close these issues.

4 participants