Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

UI and Contract support for TE2 #8

Merged

Conversation

aryamaanthakur
Copy link
Contributor

@aryamaanthakur aryamaanthakur commented Sep 3, 2024

Resolves: #3

This PR contains 2 features:

  • Include TE2 option in web ui
  • Include support for simple contract files

Additionally one more change was done to ensure passing of all the test cases - here. Because super glue dataset requires trust_remote_code set to True.

image

Signed-off-by: aryamaanthakur <aryamaan.thakur@outlook.com>
Signed-off-by: aryamaanthakur <aryamaan.thakur@outlook.com>
Signed-off-by: aryamaanthakur <aryamaan.thakur@outlook.com>
@aryamaanthakur aryamaanthakur changed the title feat(te): enable te2 in ui UI and Contract support for TE2 Sep 4, 2024
@ChanderG
Copy link
Collaborator

@aryamaanthakur Looks good. A few more things to add:

  1. Create a tool to export contract for a dataset. This can be along the lines of the train tool here: https://github.com/foundation-model-stack/fm-training-estimator/tree/main/fm_training_estimator/regressor/xgboost
  2. Instead of using the LinearReg from Sklearn, can you use the internal wrapper around the xgboost tool here?
  3. Once we have 1, we need a way to let the user input the contract in the cli/ui instead of the real dataset.

@ChanderG ChanderG merged commit 145a2a6 into foundation-model-stack:feat-token-te2 Nov 7, 2024
1 check passed
# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants