Revert "[bugfix] Fix PPMissingLayer AttributeError when Using PP" (#21189)

This commit is contained in:
Cheng Wan
2026-03-22 23:28:36 -07:00
committed by GitHub
parent 13f4f010d8
commit d6b12c401c

View File

@@ -59,28 +59,12 @@ class PPMissingLayer(torch.nn.Identity):
# https://github.com/vllm-project/vllm/blob/18ed3132d2bfe1df9a74729457b69243955221e8/vllm/model_executor/models/utils.py#L468C1-L486C1
"""
A placeholder layer for missing layers in a pipeline parallel model.
Attribute access on this placeholder returns 'self' so that common
patterns like ``isinstance(layer.mlp, SomeMoE)`` safely evaluate to
``False`` instead of raising ``AttributeError``.
"""
def __init__(self, *args, **kwargs):
super().__init__()
self.return_tuple = kwargs.get("return_tuple", False)
def __getattr__(self, name: str):
try:
return super().__getattr__(name)
except (AttributeError, KeyError):
return self
def __setattr__(self, name: str, value):
if name == "return_tuple" or name.startswith("_"):
super().__setattr__(name, value)
else:
pass
def forward(self, *args, **kwargs):
"""
Return the first arg from args or the first value from kwargs.