Sidney233 4 сар өмнө
parent
commit
c303fe66e2

+ 2 - 2
tests/unittest/test_e2e.py

@@ -204,7 +204,7 @@ class TestE2E:
             if doc_path.suffix in pdf_suffixes + image_suffixes:
                 doc_path_list.append(doc_path)
 
-        os.environ["MINERU_MODEL_SOURCE"] = "modelscope"
+        # os.environ["MINERU_MODEL_SOURCE"] = "modelscope"
         parse_doc(doc_path_list, output_dir)
 
     def test_vlm_transformers_with_default_config(self):
@@ -332,5 +332,5 @@ class TestE2E:
             if doc_path.suffix in pdf_suffixes + image_suffixes:
                 doc_path_list.append(doc_path)
 
-        os.environ["MINERU_MODEL_SOURCE"] = "modelscope"
+        # os.environ["MINERU_MODEL_SOURCE"] = "modelscope"
         parse_doc(doc_path_list, output_dir)