This commit is contained in:
unknown
2025-12-26 06:15:12 -08:00
parent 54b598a70c
commit 7987ecb3a0
3 changed files with 16 additions and 7 deletions

10
app.py
View File

@@ -105,20 +105,20 @@ Docker run image:
CPU:
docker run --rm -it -p 7860:7860 ebook2audiobook:cpu
CUDA:
docker run --gpus all --rm -it -p 7860:7860 ebook2audiobook:cu[118/121/128 etc..]
docker run --gpus all --rm -it -p 7860:7860 ebook2audiobook:cu[118/122/124/126 etc..]
ROCM:
docker run --device=/dev/kfd --device=/dev/dri --rm -it -p 7860:7860 ebook2audiobook:rocm[5.5/6.1/6.4 etc..]
docker run --device=/dev/kfd --device=/dev/dri --rm -it -p 7860:7860 ebook2audiobook:rocm[6.0/6.1/6.4 etc..]
XPU:
docker run --device=/dev/dri --rm -it -p 7860:7860 ebook2audiobook:xpu
JETSON:
docker run --runtime nvidia --rm -it -p 7860:7860 ebook2audiobook:jetson[51/60/61 etc...]
docker run --runtime nvidia --rm -it -p 7860:7860 ebook2audiobook:jetson[60/61 etc...]
Headless mode:
CPU:
docker run --rm -it -v "/my/real/ebooks/folder/absolute/path:/app/ebooks" -v "/my/real/output/folder/absolute/path:/app/audiobooks" -p 7860:7860 ebook2audiobook:cpu --headless --ebook "/app/ebooks/myfile.pdf" [--voice /app/my/voicepath/voice.mp3 etc..]
CUDA:
docker run --gpus all --rm -it -v "/my/real/ebooks/folder/absolute/path:/app/ebooks" -v "/my/real/output/folder/absolute/path:/app/audiobooks" -p 7860:7860 ebook2audiobook:cu[118/121/128 etc..] --headless --ebook "/app/ebooks/myfile.pdf" [--voice /app/my/voicepath/voice.mp3 etc..]
docker run --gpus all --rm -it -v "/my/real/ebooks/folder/absolute/path:/app/ebooks" -v "/my/real/output/folder/absolute/path:/app/audiobooks" -p 7860:7860 ebook2audiobook:cu[118/122/124/126 etc..] --headless --ebook "/app/ebooks/myfile.pdf" [--voice /app/my/voicepath/voice.mp3 etc..]
ROCM:
docker run --device=/dev/kfd --device=/dev/dri --rm -it -v "/my/real/ebooks/folder/absolute/path:/app/ebooks" -v "/my/real/output/folder/absolute/path:/app/audiobooks" -p 7860:7860 ebook2audiobook:rocm[5.5/6.1/6.4 etc..] --headless --ebook "/app/ebooks/myfile.pdf" [--voice /app/my/voicepath/voice.mp3 etc..]
docker run --device=/dev/kfd --device=/dev/dri --rm -it -v "/my/real/ebooks/folder/absolute/path:/app/ebooks" -v "/my/real/output/folder/absolute/path:/app/audiobooks" -p 7860:7860 ebook2audiobook:rocm[6.0/6.1/6.4 etc..] --headless --ebook "/app/ebooks/myfile.pdf" [--voice /app/my/voicepath/voice.mp3 etc..]
XPU:
docker run --device=/dev/dri --rm -it -v "/my/real/ebooks/folder/absolute/path:/app/ebooks" -v "/my/real/output/folder/absolute/path:/app/audiobooks" -p 7860:7860 ebook2audiobook:xpu --headless --ebook "/app/ebooks/myfile.pdf" [--voice /app/my/voicepath/voice.mp3 etc..]
JETSON:

View File

@@ -81,6 +81,8 @@ set "BROWSER_HELPER=%SCRIPT_DIR%\.bh.ps1"
set "HELP_FOUND=%ARGS:--help=%"
set "HEADLESS_FOUND=%ARGS:--headless=%"
IF NOT DEFINED DEVICE_TAG SET "DEVICE_TAG="
set "OK_SCOOP=0"
set "OK_CONDA=0"
set "OK_PROGRAMS=0"
@@ -526,7 +528,13 @@ exit /b 0
:build_docker_image
set "ARG=%~1"
for /f %%A in ('powershell -NoLogo -Command "(ConvertFrom-Json ''%ARG%'').tag"') do set "TAG=%%A"
if "%DEVICE_TAG%"=="" (
for /f "usebackq delims=" %%A in (`powershell -NoLogo -Command "(ConvertFrom-Json '%ARG%').tag"`) do (
set "TAG=%%A"
)
) else (
set "TAG=%DEVICE_TAG%"
)
"%PS_EXE%" %PS_ARGS% -command "if (!(Get-Command docker -ErrorAction SilentlyContinue)) { Write-Host '=============== Error: Docker must be installed and running!' -ForegroundColor Red; exit 1 }"
if errorlevel 1 exit /b 1
"%PS_EXE%" %PS_ARGS% -command "if (docker compose version > $null 2>&1) { exit 0 } else { exit 1 }"

View File

@@ -63,6 +63,7 @@ PACK_MGR_OPTIONS=""
BUILD_NAME=""
ISO3_LANG="eng"
SUDO="sudo"
DEVICE_TAG="${DEVICE_TAG:-}"
ARGS=("$@")
@@ -719,7 +720,7 @@ function build_docker_image {
echo -e "\e[31m=============== Error: Docker must be installed and running!.\e[0m"
return 1
fi
local TAG=$(python3 -c 'import json,sys; print(json.loads(sys.argv[1])["tag"])' "$ARG")
local TAG=${DEVICE_TAG:-$(python3 -c 'import json,sys; print(json.loads(sys.argv[1])["tag"])' "$ARG")}
local cmd_options=""
local cmd_extra=""
local py_vers="$PYTHON_VERSION"