[Model] Qwen3ForPooledOutput: forward get_input_embeddings to inner model (#23434)

2026-04-29 12:25:06 -07:00
parent 3272af2f00
commit db84a8ebbb
1 changed files with 3 additions and 0 deletions
@@ -55,6 +55,9 @@ class Qwen3ForPooledOutput(nn.Module):
        self.eos_token_id = config.eos_token_id
        # Subclasses must set self.score and self.pooler

+    def get_input_embeddings(self) -> nn.Embedding:
+        return self.model.get_input_embeddings()
+
    @torch.no_grad()
    def forward(
        self,