Support gemma2 (#1248)

modelscope · Jun 28, 2024 · d31d889 · d31d889
1 parent bd420f2
commit d31d889
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/requirements/framework.txt b/requirements/framework.txt
@@ -15,6 +15,6 @@ rouge
 safetensors
 tensorboard
 tqdm
-transformers>=4.33,<4.42
+transformers>=4.33,<4.43
 transformers_stream_generator
 trl>=0.9.4
diff --git a/swift/llm/utils/template.py b/swift/llm/utils/template.py
@@ -279,13 +279,13 @@ def add_default_tags(self, example: Dict[str, Any]) -> None:
 
     def encode(self, example: Dict[str, Any]) -> Tuple[Dict[str, Any], Dict[str, Any]]:
         """return: inputs, tokenizer_kwargs"""
-        example = example.copy()
         if not self._is_init:
             raise ValueError(
                 'Template is not initialized, please use the `get_template` function to obtain the template.')
         if example.get('images') and not isinstance(example['images'], (tuple, list)):
             # change images field to list
             example['images'] = [example['images']]
+        example = example.copy()
         self.add_default_tags(example)
         self.check_example(example)
         if example.get('objects') and isinstance(example['objects'], str):