Skip to content

Conversation

ctr-pmuruganTT
Copy link
Contributor

Ticket

closes #1045

Problem description

update Distilbert and Electra model JAX tests to use ModelLoader & ModelVariant from tt-forge-models

Checklist

  • New/Existing tests provide coverage for changes

PFA logs for reference :
distilbert.log
elec_base_discrim.log
elec_base_generator.log
elec_large_discrim.log
elec_small_discrim.log

@codecov-commenter
Copy link

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 68.41%. Comparing base (97d8592) to head (2d9d76d).
⚠️ Report is 5 commits behind head on main.

Additional details and impacted files
@@           Coverage Diff           @@
##             main    #1050   +/-   ##
=======================================
  Coverage   68.41%   68.41%           
=======================================
  Files          25       25           
  Lines        1757     1757           
=======================================
  Hits         1202     1202           
  Misses        555      555           

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

@ctr-pmuruganTT ctr-pmuruganTT force-pushed the pmurugan/update_jax_tests_for_electra_and_distilbert branch from 2d9d76d to d199bba Compare August 28, 2025 04:00
@ctr-pmuruganTT ctr-pmuruganTT marked this pull request as ready for review August 28, 2025 04:00
@ctr-pmuruganTT ctr-pmuruganTT force-pushed the pmurugan/update_jax_tests_for_electra_and_distilbert branch from 316f693 to b4f3c76 Compare August 28, 2025 07:07
@kmabeeTT
Copy link
Contributor

I approve of this, especially the tt-forge-models uplift part that I am looking forward to - thanks!

@ctr-pmuruganTT ctr-pmuruganTT force-pushed the pmurugan/update_jax_tests_for_electra_and_distilbert branch from b4f3c76 to 655aac2 Compare August 29, 2025 04:31
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.

Bringup Jax Models of Electra and Distilbert in tt-forge-models
3 participants