fix(docker): replace python3 -c with heredoc scripts to fix IndentationError

This commit is contained in:
Your Name 2026-01-25 13:43:49 +08:00
parent a9b1fc5699
commit 906f112999
4 changed files with 117 additions and 63 deletions

View file

@ -0,0 +1,36 @@
#!/bin/bash
# ==============================================================================
# MinerU 模型下載腳本
# 用於 Docker build 階段下載 PDF-Extract-Kit-1.0
# ==============================================================================
set -e
ARCH=$(uname -m)
MODELS_DIR="${MINERU_MODELS_DIR:-/opt/convertx/models/mineru}"
if [ "$ARCH" = "aarch64" ]; then
echo "⚠️ ARM64跳過 MinerU 模型下載"
exit 0
fi
echo "📦 下載 MinerU PDF-Extract-Kit-1.0 到 ${MODELS_DIR}..."
python3 <<'PYTHON'
from huggingface_hub import snapshot_download
import os
models_dir = os.environ.get('MINERU_MODELS_DIR', '/opt/convertx/models/mineru')
print(f'下載 PDF-Extract-Kit-1.0 到 {models_dir}...')
snapshot_download(
repo_id='opendatalab/PDF-Extract-Kit-1.0',
local_dir=f'{models_dir}/PDF-Extract-Kit-1.0',
local_dir_use_symlinks=False,
resume_download=True
)
print('✅ PDF-Extract-Kit-1.0 下載完成')
PYTHON
echo "✅ MinerU 模型下載完成"