lmms-lab
/

Aero-1-Audio

Text Generation

Model card Files Files and versions Community

kcz358 commited on Apr 27

Commit

ec706eb

·

verified ·

1 Parent(s): 498562e

Update processing_aero.py

Files changed (1) hide show

processing_aero.py +1 -1

processing_aero.py CHANGED Viewed

@@ -222,7 +222,7 @@ class AeroProcessor(ProcessorMixin):
                     "{{ message['content'] }}<|im_end|>\n"
                 "{% else %}"
                     "{% for content in message['content'] %}"
-                        "{% if 'audio' in content or 'audio_url' in content %}"
                             "{% set audio_count.value = audio_count.value + 1 %}"
                             "<|AUDIO|>\n"
                         "{% elif 'text' in content %}"

                     "{{ message['content'] }}<|im_end|>\n"
                 "{% else %}"
                     "{% for content in message['content'] %}"
+                        "{% if 'audio' in content or 'audio_url' in content or content['type'] == 'audio'%}"
                             "{% set audio_count.value = audio_count.value + 1 %}"
                             "<|AUDIO|>\n"
                         "{% elif 'text' in content %}"