Skip to content

Commit 8863852

Browse files
committed
add gemma in sft data convert
1 parent d4f09c0 commit 8863852

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

llm/data.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ def get_convert_example(model):
4444

4545
if base_model_prefix == "chatglm":
4646
return convert_example_chatglm
47-
elif base_model_prefix in ["chatglm_v2", "llama", "bloom", "opt", "qwen", "mixtral"]:
47+
elif base_model_prefix in ["chatglm_v2", "llama", "bloom", "opt", "qwen", "mixtral", "gemma"]:
4848
return convert_example_common
4949
else:
5050
raise ValueError(

0 commit comments

Comments
 (0)