Skip to content

[https://nvbugs/6050483][fix] pin diffusers version#13242

Merged
chang-l merged 11 commits into
NVIDIA:mainfrom
o-stoner:user/o-stoner/visual-gen-pin-diffusers
May 13, 2026
Merged

[https://nvbugs/6050483][fix] pin diffusers version#13242
chang-l merged 11 commits into
NVIDIA:mainfrom
o-stoner:user/o-stoner/visual-gen-pin-diffusers

Conversation

@o-stoner
Copy link
Copy Markdown
Collaborator

@o-stoner o-stoner commented Apr 20, 2026

Summary by CodeRabbit

  • Chores

    • Pinned diffusers package dependency to version 0.37.1 to ensure consistent and predictable package behavior.
  • Tests

    • Re-enabled previously skipped visual generation tests, reflecting improved test coverage and stability.

Description

We need to pin a Diffusers version per https://nvbugspro.nvidia.com/bug/6050483. We previously used pip install git+https://github.com/huggingface/diffusers.git in VisualGen, but that caused waived tests which we are aiming to bring back with this PR along with a pinned Diffusers version.

Test Coverage

PR Checklist

Please review the following before submitting your PR:

  • PR description clearly explains what and why. If using CodeRabbit's summary, please make sure it makes sense.

  • PR Follows TRT-LLM CODING GUIDELINES to the best of your knowledge.

  • Test cases are provided for new code paths (see test instructions)

  • Any new dependencies have been scanned for license and vulnerabilities

  • CODEOWNERS updated if ownership changes

  • Documentation updated as needed

  • Update tava architecture diagram if there is a significant design change in PR.

  • The reviewers assigned automatically/manually are appropriate for the PR.

  • Please check this after reviewing the above items as appropriate for this PR.

GitHub Bot Help

To see a list of available CI bot commands, please comment /bot help.

@o-stoner
Copy link
Copy Markdown
Collaborator Author

/bot run --disable-fail-fast

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #44518 [ run ] triggered by Bot. Commit: bac533f Link to invocation

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #44518 [ run ] completed with state SUCCESS. Commit: bac533f
/LLM/main/L0_MergeRequest_PR pipeline #34917 completed with status: 'FAILURE'

CI Report

⚠️ Action Required:

  • Please check the failed tests and fix your PR
  • If you cannot view the failures, ask the CI triggerer to share details
  • Once fixed, request an NVIDIA team member to trigger CI again

Link to invocation

@o-stoner
Copy link
Copy Markdown
Collaborator Author

/bot run --disable-fail-fast

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #44570 [ run ] triggered by Bot. Commit: bac533f Link to invocation

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #44570 [ run ] completed with state SUCCESS. Commit: bac533f
/LLM/main/L0_MergeRequest_PR pipeline #34957 completed with status: 'SUCCESS'
Pipeline passed with automatic retried tests. Check the rerun report for details.

CI Report

Link to invocation

@o-stoner o-stoner marked this pull request as ready for review April 21, 2026 16:52
@o-stoner o-stoner requested review from a team as code owners April 21, 2026 16:52
@o-stoner o-stoner requested review from chang-l and zhenhuaw-me April 21, 2026 16:52
@coderabbitai
Copy link
Copy Markdown
Contributor

coderabbitai Bot commented Apr 21, 2026

📝 Walkthrough

Walkthrough

The diffusers package dependency was pinned to version 0.37.1 in requirements.txt. The corresponding test dependency installation for diffusers was removed from test fixtures, and six visual generation test waive entries were removed from the test list.

Changes

Cohort / File(s) Summary
Dependency Configuration
requirements.txt
Updated diffusers constraint from minimum version (>=0.27.0) to exact pinned version (==0.37.1).
Test Infrastructure
tests/integration/defs/examples/test_visual_gen.py, tests/integration/test_lists/waives.txt
Removed pip install diffusers>=0.37.0 from session-scoped dependency setup; removed six visual generation test SKIP waive entries.

Estimated code review effort

🎯 2 (Simple) | ⏱️ ~5 minutes

🚥 Pre-merge checks | ✅ 4 | ❌ 1

❌ Failed checks (1 inconclusive)

Check name Status Explanation Resolution
Description check ❓ Inconclusive The description explains the issue (pinning diffusers per NVBugs 6050483) and context (previous git install caused waived tests), but the Test Coverage section is empty, which is a required section per the template. Complete the Test Coverage section by listing the specific tests (e.g., visual generation tests) that validate the pinned diffusers version and ensure waived tests are re-enabled.
✅ Passed checks (4 passed)
Check name Status Explanation
Title check ✅ Passed The title clearly identifies the main change: pinning the diffusers version to address NVBugs 6050483, which aligns with the actual modifications across requirements.txt and test configuration files.
Docstring Coverage ✅ Passed No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check.
Linked Issues check ✅ Passed Check skipped because no linked issues were found for this pull request.
Out of Scope Changes check ✅ Passed Check skipped because no linked issues were found for this pull request.

✏️ Tip: You can configure your own custom pre-merge checks in the settings.

✨ Finishing Touches
🧪 Generate unit tests (beta)
  • Create PR with unit tests

Comment @coderabbitai help to get the list of available commands and usage tips.

Copy link
Copy Markdown
Contributor

@coderabbitai coderabbitai Bot left a comment

Choose a reason for hiding this comment

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

Actionable comments posted: 1

🤖 Prompt for all review comments with AI agents
Verify each finding against the current code and only fix it if needed.

Inline comments:
In `@requirements.txt`:
- Line 7: The main requirements.txt pins diffusers==0.37.1 but the mmdit
example's requirements file currently has diffusers>=0.32.2, causing a potential
mismatch when the example is used in isolation; update the mmdit example's
requirements (the mmdit example requirements file) to either pin diffusers to
the exact same version diffusers==0.37.1 or constrain it with an upper bound
such as diffusers>=0.32.2,<=0.37.1 so the diffusers package remains compatible
with the project's use of diffusers internals.
🪄 Autofix (Beta)

Fix all unresolved CodeRabbit comments on this PR:

  • Push a commit to this branch (recommended)
  • Create a new PR with the fixes

ℹ️ Review info
⚙️ Run configuration

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Pro Plus

Run ID: d10698ce-406c-410d-a549-e7cd296f8cfd

📥 Commits

Reviewing files that changed from the base of the PR and between 7b84136 and bac533f.

📒 Files selected for processing (3)
  • requirements.txt
  • tests/integration/defs/examples/test_visual_gen.py
  • tests/integration/test_lists/waives.txt
💤 Files with no reviewable changes (2)
  • tests/integration/defs/examples/test_visual_gen.py
  • tests/integration/test_lists/waives.txt

Comment thread requirements.txt Outdated
@o-stoner o-stoner requested a review from a team as a code owner April 21, 2026 17:24
@o-stoner o-stoner requested a review from venkywonka April 21, 2026 17:24
@o-stoner
Copy link
Copy Markdown
Collaborator Author

/bot run --disable-fail-fast

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #44778 [ run ] triggered by Bot. Commit: 943dc1e Link to invocation

Copy link
Copy Markdown
Member

@zhenhuaw-me zhenhuaw-me left a comment

Choose a reason for hiding this comment

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

LGTM in general. Thanks!

Comment thread examples/visual_gen/serve/benchmark_visual_gen.sh Outdated
Comment thread examples/visual_gen/serve/benchmark_visual_gen.sh Outdated
Comment thread examples/visual_gen/serve/README.md Outdated
Comment thread tests/integration/defs/examples/test_visual_gen.py
Comment thread requirements.txt Outdated
@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #44778 [ run ] completed with state SUCCESS. Commit: 943dc1e
/LLM/main/L0_MergeRequest_PR pipeline #35133 completed with status: 'FAILURE'

CI Report

⚠️ Action Required:

  • Please check the failed tests and fix your PR
  • If you cannot view the failures, ask the CI triggerer to share details
  • Once fixed, request an NVIDIA team member to trigger CI again

Link to invocation

Signed-off-by: Olivia Stoner <245287810+o-stoner@users.noreply.github.com>
Signed-off-by: Olivia Stoner <245287810+o-stoner@users.noreply.github.com>
@o-stoner o-stoner force-pushed the user/o-stoner/visual-gen-pin-diffusers branch from 943dc1e to a1c46a1 Compare April 22, 2026 17:54
o-stoner and others added 4 commits April 22, 2026 10:55
Co-authored-by: Zhenhua Wang <4936589+zhenhuaw-me@users.noreply.github.com>
Signed-off-by: o-stoner <245287810+o-stoner@users.noreply.github.com>
Co-authored-by: Zhenhua Wang <4936589+zhenhuaw-me@users.noreply.github.com>
Signed-off-by: o-stoner <245287810+o-stoner@users.noreply.github.com>
Co-authored-by: Zhenhua Wang <4936589+zhenhuaw-me@users.noreply.github.com>
Signed-off-by: o-stoner <245287810+o-stoner@users.noreply.github.com>
Signed-off-by: Olivia Stoner <245287810+o-stoner@users.noreply.github.com>
@o-stoner
Copy link
Copy Markdown
Collaborator Author

/bot run --disable-fail-fast

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #45004 [ run ] triggered by Bot. Commit: 70f2887 Link to invocation

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #45004 [ run ] completed with state SUCCESS. Commit: 70f2887
/LLM/main/L0_MergeRequest_PR pipeline #35322 completed with status: 'SUCCESS'
Pipeline passed with automatic retried tests. Check the rerun report for details.

CI Report

Link to invocation

Signed-off-by: Zhenhua Wang <4936589+zhenhuaw-me@users.noreply.github.com>
Comment thread tests/integration/test_lists/waives.txt Outdated
Signed-off-by: Zhenhua Wang <4936589+zhenhuaw-me@users.noreply.github.com>
@zhenhuaw-me
Copy link
Copy Markdown
Member

/bot run --disable-fail-fast --add-multi-gpu-test

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #45671 [ run ] triggered by Bot. Commit: c61ef75 Link to invocation

@chang-l chang-l enabled auto-merge (squash) April 27, 2026 17:49
@chang-l
Copy link
Copy Markdown
Collaborator

chang-l commented Apr 27, 2026

/bot run

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #45798 [ run ] triggered by Bot. Commit: c61ef75 Link to invocation

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #45798 [ run ] completed with state FAILURE. Commit: c61ef75
/LLM/main/L0_MergeRequest_PR pipeline #35988 completed with status: 'ABORTED'

CI Report

⚠️ Action Required:

  • Please check the failed tests and fix your PR
  • If you cannot view the failures, ask the CI triggerer to share details
  • Once fixed, request an NVIDIA team member to trigger CI again

Link to invocation

Signed-off-by: o-stoner <245287810+o-stoner@users.noreply.github.com>
@o-stoner
Copy link
Copy Markdown
Collaborator Author

o-stoner commented May 4, 2026

/bot run --disable-fail-fast --add-multi-gpu-test

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #46668 [ run ] triggered by Bot. Commit: 98fad0a Link to invocation

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #46668 [ run ] completed with state SUCCESS. Commit: 98fad0a
/LLM/main/L0_MergeRequest_PR pipeline #36710 completed with status: 'FAILURE'

CI Report

⚠️ Action Required:

  • Please check the failed tests and fix your PR
  • If you cannot view the failures, ask the CI triggerer to share details
  • Once fixed, request an NVIDIA team member to trigger CI again

CI Agent Failure Analysis

Link to invocation

@o-stoner
Copy link
Copy Markdown
Collaborator Author

o-stoner commented May 5, 2026

/bot run --disable-fail-fast --add-multi-gpu-test

Signed-off-by: o-stoner <245287810+o-stoner@users.noreply.github.com>
@o-stoner
Copy link
Copy Markdown
Collaborator Author

o-stoner commented May 5, 2026

/bot run --disable-fail-fast --add-multi-gpu-test

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #46838 [ run ] triggered by Bot. Commit: 30da507 Link to invocation

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #46838 [ run ] completed with state ABORTED. Commit: 30da507

Link to invocation

Signed-off-by: o-stoner <245287810+o-stoner@users.noreply.github.com>
@o-stoner
Copy link
Copy Markdown
Collaborator Author

/bot run --disable-fail-fast --add-multi-gpu-test

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #47621 [ run ] triggered by Bot. Commit: 431a973 Link to invocation

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #47621 [ run ] completed with state SUCCESS. Commit: 431a973
/LLM/main/L0_MergeRequest_PR pipeline #37526 completed with status: 'FAILURE'

CI Report

⚠️ Action Required:

  • Please check the failed tests and fix your PR
  • If you cannot view the failures, ask the CI triggerer to share details
  • Once fixed, request an NVIDIA team member to trigger CI again

CI Agent Failure Analysis

Link to invocation

@o-stoner
Copy link
Copy Markdown
Collaborator Author

/bot run --disable-fail-fast --add-multi-gpu-test

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #48022 [ run ] triggered by Bot. Commit: 431a973 Link to invocation

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #48022 [ run ] completed with state SUCCESS. Commit: 431a973
/LLM/main/L0_MergeRequest_PR pipeline #37858 completed with status: 'SUCCESS'

CI Report

Link to invocation

@chang-l chang-l merged commit ff6e1d9 into NVIDIA:main May 13, 2026
6 checks passed
yufeiwu-nv pushed a commit to yufeiwu-nv/TensorRT-LLM that referenced this pull request May 19, 2026
Signed-off-by: Olivia Stoner <245287810+o-stoner@users.noreply.github.com>
Signed-off-by: o-stoner <245287810+o-stoner@users.noreply.github.com>
Signed-off-by: Zhenhua Wang <4936589+zhenhuaw-me@users.noreply.github.com>
Co-authored-by: Zhenhua Wang <4936589+zhenhuaw-me@users.noreply.github.com>
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.

5 participants