chunhuizng commited on
Commit
cba003b
·
verified ·
1 Parent(s): 10c642f

Create modeling_audio_only_thinker.py

Browse files
Files changed (1) hide show
  1. modeling_audio_only_thinker.py +11 -0
modeling_audio_only_thinker.py ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import Qwen2_5OmniThinkerForConditionalGeneration
2
+
3
+ class AudioOnlyThinker(Qwen2_5OmniThinkerForConditionalGeneration):
4
+ def __init__(self, config):
5
+ super().__init__(config)
6
+ self.visual = None
7
+ if hasattr(self.config, "vision_config"):
8
+ del self.config.vision_config
9
+
10
+ def forward(self, *args, pixel_values=None, pixel_values_videos=None, **kwargs):
11
+ return super().forward(*args, pixel_values=None, pixel_values_videos=None, **kwargs)