diff --git a/swift/llm/utils/template.py b/swift/llm/utils/template.py
index 0e0e7b1459..cf719bee42 100644
--- 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):