We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent aeb82b1 commit 9c32df6Copy full SHA for 9c32df6
vllm/model_executor/models/adapters.py
@@ -175,9 +175,14 @@ def __init__(
175
self.vllm_config = vllm_config
176
177
# These are not used in pooling models
178
- for attr in ("lm_head", "logits_processor"):
179
- if hasattr(self, attr):
180
- delattr(self, attr)
+ objects_to_clean = [self]
+ if language_model := getattr(self, "language_model", None):
+ objects_to_clean.append(language_model)
181
+
182
+ for obj in objects_to_clean:
183
+ for attr in ("lm_head", "logits_processor"):
184
+ if hasattr(obj, attr):
185
+ delattr(obj, attr)
186
187
# If the model already defines a pooler instance, don't overwrite it
188
if not getattr(self, "pooler", None):
0 commit comments