biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[Model] Remove unnecessary get_language_model (#37545)

Browse Source 
Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
This commit is contained in:
Cyrus Leung
2026-03-19 20:00:36 +08:00
committed by GitHub
parent c7bc12c20f
commit 7a6ebcbfcf
7 changed files with 93 additions and 95 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
vllm/model_executor/models/fireredasr2.py
View File

@@ -754,12 +754,17 @@ class FireRedASR2ForConditionalGeneration(
self.config = config
self.dtype = vllm_config.model_config.dtype
self.model = FireRedASR2Model(
vllm_config=vllm_config,
prefix=maybe_prefix(prefix, "model"),
)
logit_scale = getattr(config, "logit_scale", 1.0)
with self._mark_composite_model(
vllm_config,
language_targets=Qwen2ForCausalLM,
tower_targets={"audio": (FireRedASR2Encoder, FireRedASR2Adapter)},
):
self.model = FireRedASR2Model(
vllm_config=vllm_config,
prefix=maybe_prefix(prefix, "model"),
)
logit_scale = getattr(config, "logit_scale", 1.0)
self.logits_processor = LogitsProcessor(config.vocab_size, scale=logit_scale)
def forward(