Merge branch 'main' into release/2.5

modelscope · Oct 21, 2024 · 64d8080 · 64d8080
2 parents 6e49bf6 + 1756257
commit 64d8080
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/swift/llm/utils/template.py b/swift/llm/utils/template.py
@@ -3173,7 +3173,7 @@ def data_collator(self, batch: List[Dict[str, Any]], padding_to: Optional[int] =
         return res
 
 
-register_template(TemplateType.molmo, MolmoTemplate())
+register_template(TemplateType.molmo, MolmoTemplate(), lazy_tokenize=True, use_model=True)
 
 
 class PaliGemmaTemplate(Template):