Преглед изворни кода

fix:'title_aided_config' is not defined

myhloli пре 4 месеци
родитељ
комит
92d43c7b23
1 измењених фајлова са 2 додато и 1 уклоњено
  1. 2 1
      mineru/backend/vlm/token_to_middle_json.py

+ 2 - 1
mineru/backend/vlm/token_to_middle_json.py

@@ -12,7 +12,8 @@ from mineru.version import __version__
 
 heading_level_import_success = False
 llm_aided_config = get_llm_aided_config()
-if llm_aided_config and llm_aided_config.get('title_aided', {}).get('enable', False):
+title_aided_config = llm_aided_config.get('title_aided', {})
+if llm_aided_config and title_aided_config.get('enable', False):
     try:
         from mineru.utils.llm_aided import llm_aided_title
         from mineru.backend.pipeline.model_init import AtomModelSingleton