Skip to content

Commit 781fb2c

Browse files
committed
security(peft): enforce torch.nn.init prefix validation
Signed-off-by: RinZ27 <222222878+RinZ27@users.noreply.github.com>
1 parent e4df91b commit 781fb2c

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

modelopt/torch/peft/config.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -92,6 +92,10 @@ class PEFTAttributeConfig(ModeloptBaseConfig):
9292
@classmethod
9393
def _parse_init_callable(cls, v):
9494
if isinstance(v, str):
95+
if not v.startswith("torch.nn.init."):
96+
raise ValueError(
97+
f"Initializer must be from 'torch.nn.init', got '{v}'"
98+
)
9599
try:
96100
module_path, func_name = v.rsplit(".", 1)
97101
mod = importlib.import_module(module_path)

0 commit comments

Comments
 (0)