Skip to content

Add inference test config for boltz2 model #2267

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.

Sign up for GitHub

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

Jump to bottom
Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Add inference test config for boltz2 model #2267

wants to merge 1 commit into from
+11 −3

Conversation

@meenakshiramanathan1
Copy link
Contributor

@meenakshiramanathan1 meenakshiramanathan1 commented Nov 21, 2025 •
edited
Loading

Ticket

Problem description

  • Add inference test config for boltz2 pytorch model

What's changed

  • Added inference test config for boltz2 pytorch mode. This test executes for a runtime of two hours which might affect nightly pipeline hence it is skipped for now.

Logs

boltz_xla.zip

@meenakshiramanathan1 meenakshiramanathan1 force-pushed the mramanathan/boltz2 branch 3 times, most recently from af191f0 to 3a0aefa Compare November 25, 2025 04:58
@meenakshiramanathan1 meenakshiramanathan1 marked this pull request as ready for review November 25, 2025 05:59
nvukobratTT
nvukobratTT reviewed Nov 25, 2025
View reviewed changes
venv/requirements-dev.txt
# whisper and wav2vec2 models
torchcodec

#boltz2
Copy link

@nvukobratTT nvukobratTT Nov 25, 2025

Choose a reason for hiding this comment

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

These should be requirements for a specific model, not included in the general requirements for now.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@nvukobratTT nvukobratTT nvukobratTT left review comments

@saiarthiraguram saiarthiraguram saiarthiraguram approved these changes

@AleksKnezevic AleksKnezevic Awaiting requested review from AleksKnezevic AleksKnezevic is a code owner

@ajakovljevicTT ajakovljevicTT Awaiting requested review from ajakovljevicTT ajakovljevicTT is a code owner

@jameszianxuTT jameszianxuTT Awaiting requested review from jameszianxuTT jameszianxuTT is a code owner

@kmabeeTT kmabeeTT Awaiting requested review from kmabeeTT kmabeeTT is a code owner

@LPanosTT LPanosTT Awaiting requested review from LPanosTT LPanosTT is a code owner

@mrakitaTT mrakitaTT Awaiting requested review from mrakitaTT mrakitaTT is a code owner

@mstojkovicTT mstojkovicTT Awaiting requested review from mstojkovicTT mstojkovicTT is a code owner

@ndrakulicTT ndrakulicTT Awaiting requested review from ndrakulicTT ndrakulicTT is a code owner

@ppadjinTT ppadjinTT Awaiting requested review from ppadjinTT ppadjinTT is a code owner

@sdjukicTT sdjukicTT Awaiting requested review from sdjukicTT sdjukicTT is a code owner

@sgligorijevicTT sgligorijevicTT Awaiting requested review from sgligorijevicTT sgligorijevicTT is a code owner

@vkovinicTT vkovinicTT Awaiting requested review from vkovinicTT vkovinicTT is a code owner

@vzeljkovicTT vzeljkovicTT Awaiting requested review from vzeljkovicTT vzeljkovicTT is a code owner

At least 1 approving review is required to merge this pull request.

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

4 participants

@meenakshiramanathan1 @nvukobratTT @saiarthiraguram