Skip to content

Commit ac85781

Browse files
committed
Merge branch 'main' into v2.1
2 parents 09661d5 + b42326e commit ac85781

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

swift/llm/utils/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2828,7 +2828,7 @@ def _new_forward(*args, **kwargs):
28282828
ModelType.mini_internvl_chat_4b_v1_5,
28292829
'OpenGVLab/Mini-InternVL-Chat-4B-V1-5',
28302830
LoRATM.phi3,
2831-
TemplateType.internvl,
2831+
TemplateType.internvl_phi3,
28322832
requires=['transformers>=4.35', 'timm'],
28332833
support_flash_attn=True,
28342834
placeholder_tokens=['<IMG_CONTEXT>'],

swift/llm/utils/template.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -976,10 +976,11 @@ def get_generate_ids(generate_ids: Tensor, input_token_len: int) -> List[int]:
976976

977977

978978
class InternvlPhi3Template(InternvlTemplate):
979+
system = 'You are an AI assistant whose name is Phi-3.'
979980

980981
def __init__(self):
981982
Template.__init__(self, ['<s>'], ['<|user|>\n', [-100], '{{QUERY}}<|end|>\n<|assistant|>\n'], ['<|end|>\n'],
982-
['<|end|>'], None, ['<s><|system|>\n{{SYSTEM}}<|end|>\n'])
983+
['<|end|>'], self.system, ['<s><|system|>\n{{SYSTEM}}<|end|>\n'])
983984

984985

985986
register_template(

0 commit comments

Comments
 (0)