Skip to content
This repository has been archived by the owner on Oct 11, 2024. It is now read-only.

Rs/model integration tests logprobs #71

Merged
merged 51 commits into from
Feb 29, 2024

Conversation

robertgshaw2-neuralmagic
Copy link
Collaborator

@robertgshaw2-neuralmagic robertgshaw2-neuralmagic commented Feb 28, 2024

SUMMARY:
Adds end to end model tests

TEST PLAN:
Compares logprobs of results from hf model vs vllm model at fp16 and bfloat16

Copy link
Member

@andy-neuma andy-neuma left a comment

Choose a reason for hiding this comment

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

cool

@robertgshaw2-neuralmagic robertgshaw2-neuralmagic merged commit 2879d9d into main Feb 29, 2024
2 checks passed
@robertgshaw2-neuralmagic robertgshaw2-neuralmagic deleted the rs/model-integration-tests-logprobs branch February 29, 2024 22:31
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