From 59c7de05395d132cdf64cc304f85baddf3c13cb8 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Tue, 15 Sep 2026 06:12:20 -0600 Subject: [PATCH] Increase the repition penalty --- extensions_built_in/captioner/Qwen3OmniCaptioner.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/extensions_built_in/captioner/Qwen3OmniCaptioner.py b/extensions_built_in/captioner/Qwen3OmniCaptioner.py index c4ad548..6c72731 100644 --- a/extensions_built_in/captioner/Qwen3OmniCaptioner.py +++ b/extensions_built_in/captioner/Qwen3OmniCaptioner.py @@ -791,8 +791,8 @@ class Qwen3OmniCaptioner(BaseCaptioner): self.model._pad_mask_2d = inputs.get("attention_mask", None) gen_kwargs = self._gen_kwargs(inputs["input_ids"].shape[1]) if items[0]["kind"] == "audio": - # greedy lyric transcription loops on syllables; mild penalty (the model's own talker default) - gen_kwargs["repetition_penalty"] = 1.05 + # greedy lyric transcription loops on musical phrases; stronger penalty to mitigate + gen_kwargs["repetition_penalty"] = 1.20 generated_ids = self.model.generate( **inputs, use_audio_in_video=use_audio,