[Model] Qwen3ForPooledOutput: forward get_input_embeddings to inner model (#23434)
This commit is contained in:
@@ -55,6 +55,9 @@ class Qwen3ForPooledOutput(nn.Module):
|
|||||||
self.eos_token_id = config.eos_token_id
|
self.eos_token_id = config.eos_token_id
|
||||||
# Subclasses must set self.score and self.pooler
|
# Subclasses must set self.score and self.pooler
|
||||||
|
|
||||||
|
def get_input_embeddings(self) -> nn.Embedding:
|
||||||
|
return self.model.get_input_embeddings()
|
||||||
|
|
||||||
@torch.no_grad()
|
@torch.no_grad()
|
||||||
def forward(
|
def forward(
|
||||||
self,
|
self,
|
||||||
|
|||||||
Reference in New Issue
Block a user