Skip to content

Add DiT to the entry points to the model registry#1679

Closed
CharlelieLrt wants to merge 1 commit into
NVIDIA:2.1.0-rcfrom
CharlelieLrt:dit-checkpoint-bugfix
Closed

Add DiT to the entry points to the model registry#1679
CharlelieLrt wants to merge 1 commit into
NVIDIA:2.1.0-rcfrom
CharlelieLrt:dit-checkpoint-bugfix

Conversation

@CharlelieLrt
Copy link
Copy Markdown
Collaborator

PhysicsNeMo Pull Request

Description

Add the DiT to the model 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

This PR registers the DiT (Diffusion Transformer) model in the PhysicsNeMo model registry by adding a single entry-point line to pyproject.toml.

  • Adds DiT = "physicsnemo.models.dit:DiT" under [project.entry-points."physicsnemo.models"], enabling registry-based model discovery. The target module and class exist and are correctly exported from physicsnemo/models/dit/__init__.py.

Important Files Changed

Filename Overview
pyproject.toml Adds a DiT entry to the [project.entry-points."physicsnemo.models"] section; the target module physicsnemo.models.dit:DiT exists and exports DiT correctly.

Reviews (1): Last reviewed commit: "Add DiT to the entry points to the model..." | Re-trigger Greptile

@CharlelieLrt
Copy link
Copy Markdown
Collaborator Author

Replaced by #1680.

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.

1 participant