update example

InternLM · Mar 11, 2024 · 9ed42c1 · 9ed42c1
1 parent 1a4cf76
commit 9ed42c1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/examples/vl/qwen_model.py b/examples/vl/qwen_model.py
@@ -53,6 +53,9 @@ def messages2prompt(self, messages, sequence_start=True):
                        assistant=self.eoa + self.stop_word_suffix,
                        system=self.eosys)
         ret = ''
+        if self.meta_instruction is not None:
+            if len(messages) and messages[0]['role'] != 'system':
+                ret += f'{self.system}{self.meta_instruction}{self.eosys}'
         for message in messages:
             role = message['role']
             content = message['content']

diff --git a/examples/vl/xcomposer_model.py b/examples/vl/xcomposer_model.py
@@ -67,6 +67,9 @@ def messages2prompt(self, messages, sequence_start=True):
                        assistant=self.eoa + self.stop_word_suffix,
                        system=self.eosys)
         ret = ''
+        if self.meta_instruction is not None:
+            if len(messages) and messages[0]['role'] != 'system':
+                ret += f'{self.system}{self.meta_instruction}{self.eosys}'
         for message in messages:
             role = message['role']
             content = message['content']