Thanks to visit codestin.com
Credit goes to github.com

Skip to content
This repository was archived by the owner on Jun 3, 2025. It is now read-only.

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

Merged
dbogunowicz merged 15 commits into
mainfrom
feature/damian/final_llm_tests
Oct 23, 2023
Merged

[LLM Tests] Config support for LLM test suite / enable test suite in GHA#1324
dbogunowicz merged 15 commits into
mainfrom
feature/damian/final_llm_tests

Conversation

@dbogunowicz

@dbogunowicz dbogunowicz commented Oct 17, 2023

Copy link
Copy Markdown
Contributor

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
Comment thread src/deepsparse/transformers/pipelines/text_generation.py
Comment thread tests/deepsparse/transformers/pipelines/configs/gpt_neo.yaml Outdated
Comment thread tests/deepsparse/transformers/pipelines/test_text_generation.py Outdated
@dbogunowicz

Copy link
Copy Markdown
Contributor Author

Tests are green yay! :)

Comment thread tests/deepsparse/transformers/pipelines/test_text_generation.py Outdated
Comment thread tests/deepsparse/transformers/pipelines/test_chat.py Outdated
Comment thread tests/deepsparse/transformers/pipelines/test_chat.py Outdated
Comment thread tests/deepsparse/transformers/pipelines/helpers.py Outdated
Comment thread tests/deepsparse/transformers/pipelines/helpers.py Outdated
Comment thread tests/deepsparse/transformers/pipelines/helpers.py Outdated
Comment thread tests/deepsparse/transformers/pipelines/configs/gpt_neo.yaml Outdated
Comment thread tests/deepsparse/transformers/pipelines/configs/gpt_neo.yaml Outdated
Comment thread tests/deepsparse/transformers/pipelines/configs/gpt_neo.yaml Outdated
Comment thread tests/deepsparse/transformers/pipelines/configs/gpt_neo.yaml Outdated
Comment thread tests/deepsparse/transformers/pipelines/configs/gpt_neo.yaml Outdated
Comment thread tests/deepsparse/transformers/pipelines/test_text_generation.py Outdated
Comment thread tests/deepsparse/transformers/pipelines/test_text_generation.py Outdated
Comment thread tests/deepsparse/transformers/pipelines/test_text_generation.py Outdated
Comment thread tests/deepsparse/transformers/pipelines/test_text_generation.py Outdated
Comment thread tests/deepsparse/transformers/pipelines/test_text_generation.py Outdated

@Satrat Satrat left a comment

Copy link
Copy Markdown

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
@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 subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants