From db84a8ebbbc2c9c47a6bb333d5b63c621db0bc08 Mon Sep 17 00:00:00 2001
From: jsheng_Linkedin <162757399+fortunecookiee@users.noreply.github.com>
Date: Wed, 29 Apr 2026 12:25:06 -0700
Subject: [PATCH] [Model] Qwen3ForPooledOutput: forward get_input_embeddings to
 inner model (#23434)

---
 python/sglang/srt/models/qwen3_classification.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/python/sglang/srt/models/qwen3_classification.py b/python/sglang/srt/models/qwen3_classification.py
index 8afc365bc..352f7c7d8 100644
--- a/python/sglang/srt/models/qwen3_classification.py
+++ b/python/sglang/srt/models/qwen3_classification.py
@@ -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,