Merge pull request #172 from dt-yy/master

add gpu ci
2026-03-27 11:08:32 +07:00 · 2024-07-19 15:45:52 +08:00
parent 5feeb836ff 305c77cd52
commit ab1ec00232
11 changed files with 3 additions and 6 deletions
--- a/.github/workflows/gpu-ci.yml
+++ b/.github/workflows/gpu-ci.yml
@@ -31,9 +31,6 @@ jobs:
      
    - name: check-requirements
      run: |
-        echo $PATH
-        conda init
-        conda activate QA
        source ~/.bashrc
        pip install magic-pdf[full-cpu]==0.6.1 -i https://pypi.tuna.tsinghua.edu.cn/simple
        pip install https://raw.githubusercontent.com/myhloli/wheels/main/assets/whl/detectron2/detectron2-0.6-cp310-cp310-linux_x86_64.whl
--- a/magic_pdf/libs/pycache/init.cpython-312.pyc
+++ b/magic_pdf/libs/pycache/init.cpython-312.pyc
--- a/magic_pdf/libs/pycache/version.cpython-312.pyc
+++ b/magic_pdf/libs/pycache/version.cpython-312.pyc
--- a/tests/pycache/test_unit.cpython-39-pytest-7.4.0.pyc
+++ b/tests/pycache/test_unit.cpython-39-pytest-7.4.0.pyc
--- a/tests/test_cli/pycache/test_bench.cpython-39-pytest-7.4.0.pyc
+++ b/tests/test_cli/pycache/test_bench.cpython-39-pytest-7.4.0.pyc
--- a/tests/test_cli/pycache/test_cli.cpython-39-pytest-7.4.0.pyc
+++ b/tests/test_cli/pycache/test_cli.cpython-39-pytest-7.4.0.pyc
--- a/tests/test_cli/conf/pycache/conf.cpython-39.pyc
+++ b/tests/test_cli/conf/pycache/conf.cpython-39.pyc
--- a/tests/test_cli/lib/pycache/init.cpython-39.pyc
+++ b/tests/test_cli/lib/pycache/init.cpython-39.pyc
--- a/tests/test_cli/lib/pycache/calculate_score.cpython-39.pyc
+++ b/tests/test_cli/lib/pycache/calculate_score.cpython-39.pyc
--- a/tests/test_cli/lib/pycache/scoring.cpython-39.pyc
+++ b/tests/test_cli/lib/pycache/scoring.cpython-39.pyc
--- a/tests/test_cli/test_bench_gpu.py
+++ b/tests/test_cli/test_bench_gpu.py
@@ -6,7 +6,7 @@ import json
 from magic_pdf.pipe.UNIPipe import UNIPipe
 from magic_pdf.rw.DiskReaderWriter import DiskReaderWriter
 from lib import calculate_score
-
+import shutil
 pdf_res_path = conf.conf["pdf_res_path"]
 code_path = conf.conf["code_path"]
 pdf_dev_path = conf.conf["pdf_dev_path"]
@@ -58,8 +58,8 @@ def pdf_to_markdown():
        if not os.path.exists(dir_path):
            os.makedirs(dir_path, exist_ok=True)
        res_path = os.path.join(dir_path, f"{demo_name}.md")
-        #src_path = os.path.join(pdf_res_path, "pdf", f"{demo_name}.pdf") 
-        #shutil.copy(src_path, res_path)
+        src_path = os.path.join(pdf_res_path, demo_name, "auto", f"{demo_name}.md")
+        shutil.copy(src_path, res_path)