add gemma in sft data convert

Southpika · Southpika · commit 2f8b5d83f1ce · 2024-04-02T19:02:18.000+08:00
diff --git a/llm/data.py b/llm/data.py
@@ -48,7 +48,7 @@ def get_convert_example(model):
         return convert_example_common
     else:
         raise ValueError(
-            f"Unknown base_model_prefix: {model.base_model_prefix}. Supported base_model_prefix list: chatglm, bloom, llama, qwen, mixtral"
+            f"Unknown base_model_prefix: {model.base_model_prefix}. Supported base_model_prefix list: chatglm, bloom, llama, qwen, mixtral, gemma"
         )
 
 

Original file line number	Diff line number	Diff line change
`@@ -48,7 +48,7 @@ def get_convert_example(model):`
`48`	`48`	`return convert_example_common`
`49`	`49`	`else:`
`50`	`50`	`raise ValueError(`
`51`		`- f"Unknown base_model_prefix: {model.base_model_prefix}. Supported base_model_prefix list: chatglm, bloom, llama, qwen, mixtral"`
	`51`	`+ f"Unknown base_model_prefix: {model.base_model_prefix}. Supported base_model_prefix list: chatglm, bloom, llama, qwen, mixtral, gemma"`
`52`	`52`	`)`
`53`	`53`
`54`	`54`