OpenGVLab
/

VideoChat-Flash-Qwen2_5-2B_res448

Video-Text-to-Text

videochat_flash_qwen

feature-extraction

Model card Files Files and versions

lixinhao commited on Mar 4

Commit

2cc4ab3

·

verified ·

1 Parent(s): ffa7227

Update modeling_videochat_flash.py

Files changed (1) hide show

modeling_videochat_flash.py +1 -1

modeling_videochat_flash.py CHANGED Viewed

@@ -679,7 +679,7 @@ class VideoChatFlashQwenForCausalLM(LlavaMetaForCausalLM, Qwen2ForCausalLM_Flash
         outputs = tokenizer.batch_decode(output_ids, skip_special_tokens=True)[0].strip()
         if outputs.endswith(stop_str):
-                outputs = outputs[: -len(stop_str)]
         outputs = outputs.strip()

         outputs = tokenizer.batch_decode(output_ids, skip_special_tokens=True)[0].strip()
         if outputs.endswith(stop_str):
+            outputs = outputs[: -len(stop_str)]
         outputs = outputs.strip()