Skip to content

Added DiT to entry points#1680

Merged
coreyjadams merged 4 commits into
NVIDIA:mainfrom
CharlelieLrt:add-dit-to-registry
May 29, 2026
Merged

Added DiT to entry points#1680
coreyjadams merged 4 commits into
NVIDIA:mainfrom
CharlelieLrt:add-dit-to-registry

Conversation

@CharlelieLrt
Copy link
Copy Markdown
Collaborator

PhysicsNeMo Pull Request

Description

Adds DiT model to the entry-points to have it in the registry.

Checklist

Dependencies

Review Process

All PRs are reviewed by the PhysicsNeMo team before merging.

Depending on which files are changed, GitHub may automatically assign a maintainer for review.

We are also testing AI-based code review tools (e.g., Greptile), which may add automated comments with a confidence score.
This score reflects the AI’s assessment of merge readiness and is not a qualitative judgment of your work, nor is
it an indication that the PR will be accepted / rejected.

AI-generated feedback should be reviewed critically for usefulness.
You are not required to respond to every AI comment, but they are intended to help both authors and reviewers.
Please react to Greptile comments with 👍 or 👎 to provide feedback on their accuracy.

Signed-off-by: Charlelie Laurent <claurent@nvidia.com>
@copy-pr-bot
Copy link
Copy Markdown

copy-pr-bot Bot commented May 27, 2026

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@greptile-apps
Copy link
Copy Markdown
Contributor

greptile-apps Bot commented May 27, 2026

Greptile Summary

Registers the existing DiT (Diffusion Transformer) model in the physicsnemo.models entry-points group so it appears in the model registry, mirroring the pattern already used for SongUNet, DhariwalUNet, and other models.

  • Adds one line to pyproject.toml pointing DiTphysicsnemo.models.dit:DiT; the target module and class already exist and are correctly exported.

Important Files Changed

Filename Overview
pyproject.toml Adds DiT = "physicsnemo.models.dit:DiT" to the [project.entry-points."physicsnemo.models"] section; the module and class exist and are correctly exported.

Reviews (1): Last reviewed commit: "Added DiT to entry points" | Re-trigger Greptile

@CharlelieLrt CharlelieLrt enabled auto-merge May 27, 2026 20:49
@CharlelieLrt
Copy link
Copy Markdown
Collaborator Author

/blossom-ci

@CharlelieLrt
Copy link
Copy Markdown
Collaborator Author

/ok to test 191752b

@CharlelieLrt CharlelieLrt added this pull request to the merge queue May 27, 2026
@github-merge-queue github-merge-queue Bot removed this pull request from the merge queue due to no response for status checks May 28, 2026
@CharlelieLrt CharlelieLrt added this pull request to the merge queue May 28, 2026
@coreyjadams coreyjadams removed this pull request from the merge queue due to a manual request May 28, 2026
@coreyjadams coreyjadams enabled auto-merge May 28, 2026 00:53
@coreyjadams
Copy link
Copy Markdown
Collaborator

/blossom-ci

@coreyjadams
Copy link
Copy Markdown
Collaborator

/ok to test 98693e1

@coreyjadams
Copy link
Copy Markdown
Collaborator

/blossom-ci

@coreyjadams
Copy link
Copy Markdown
Collaborator

/blossom-co

@coreyjadams
Copy link
Copy Markdown
Collaborator

/blossom-ci

2 similar comments
@CharlelieLrt
Copy link
Copy Markdown
Collaborator Author

/blossom-ci

@CharlelieLrt
Copy link
Copy Markdown
Collaborator Author

/blossom-ci

@CharlelieLrt
Copy link
Copy Markdown
Collaborator Author

/blossom-ci

@CharlelieLrt
Copy link
Copy Markdown
Collaborator Author

/ok to test 98693e1

@copy-pr-bot
Copy link
Copy Markdown

copy-pr-bot Bot commented May 29, 2026

/ok to test 98693e1

@CharlelieLrt, there was an error processing your request: E2

See the following link for more information: https://docs.gha-runners.nvidia.com/cpr/e/2/

@CharlelieLrt
Copy link
Copy Markdown
Collaborator Author

/ok to test 560f4c2

@coreyjadams coreyjadams added this pull request to the merge queue May 29, 2026
Merged via the queue into NVIDIA:main with commit d8a8380 May 29, 2026
6 checks passed
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.

3 participants