Skip to content

Pre-initialize torch._dynamo to prevent double-registration with peft torch.compile() call #1228

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

Merged
hychiang-git merged 7 commits into from
Apr 11, 2026
+14 −1
Merged

Pre-initialize torch._dynamo to prevent double-registration with peft torch.compile() call #1228

Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
203e219
Pre-initialize torch._dynamo to prevent double-registration with pef…
hychiang-git Apr 9, 2026
8e1696e
Use importlib to avoid polluting the namesapce, and fix ruff check
hychiang-git Apr 9, 2026
566917a
Merge branch 'main' into hungyuehc/fix_local_run
hychiang-git Apr 10, 2026
88f4d17
Merge branch 'main' into hungyuehc/fix_local_run
shengliangxu Apr 10, 2026
d232042
Merge branch 'main' into hungyuehc/fix_local_run
hychiang-git Apr 10, 2026
25d6f16
Merge branch 'main' into hungyuehc/fix_local_run
hychiang-git Apr 10, 2026
89b860d
Merge branch 'main' into hungyuehc/fix_local_run
hychiang-git Apr 11, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 14 additions & 1 deletion modelopt/torch/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -15,12 +15,25 @@

"""Model optimization and deployment subpackage for torch."""

import importlib
import warnings as _warnings

from packaging.version import Version as _Version
from torch import __version__ as _torch_version

from . import distill, nas, opt, peft, prune, quantization, sparsity, speculative, utils
# Pre-initialize torch._dynamo to prevent double-registration with peft's torch.compile() call
importlib.import_module("torch._dynamo")
from . import ( # noqa: E402
distill,
nas,
opt,
peft,
prune,
quantization,
sparsity,
speculative,
utils,
)

if _Version(_torch_version) < _Version("2.9"):
_warnings.warn(
Expand Down
Loading