Feat: Rework multimodal support (mllama, llava, pixtral, qwen2, qwen25, gemma3, mistral3) (#2435)
This commit is contained in:
@@ -58,7 +58,7 @@ class TestGemma3Text:
|
||||
"bos_token": "<bos>",
|
||||
"eos_token": "<eos>",
|
||||
},
|
||||
"chat_template": "gemma3_text",
|
||||
"chat_template": "gemma3",
|
||||
"num_epochs": 1,
|
||||
"micro_batch_size": 1,
|
||||
"gradient_accumulation_steps": 4,
|
||||
@@ -105,7 +105,7 @@ class TestGemma3Text:
|
||||
"split": "train[:1%]",
|
||||
},
|
||||
],
|
||||
"chat_template": "gemma3_text",
|
||||
"chat_template": "gemma3",
|
||||
"special_tokens": {
|
||||
"bos_token": "<bos>",
|
||||
"eos_token": "<eos>",
|
||||
|
||||
Reference in New Issue
Block a user