59 lines
1.7 KiB
Bash
Executable File
59 lines
1.7 KiB
Bash
Executable File
#!/bin/bash
|
|
set -e
|
|
|
|
# Configuration
|
|
REGISTRY="code.foss.global"
|
|
NAMESPACE="host.today"
|
|
IMAGE_NAME="ht-docker-ai"
|
|
|
|
# Colors for output
|
|
GREEN='\033[0;32m'
|
|
BLUE='\033[0;34m'
|
|
NC='\033[0m' # No Color
|
|
|
|
echo -e "${BLUE}Building ht-docker-ai images...${NC}"
|
|
|
|
# Build GPU variant
|
|
echo -e "${GREEN}Building MiniCPM-V 4.5 GPU variant...${NC}"
|
|
docker build \
|
|
-f Dockerfile_minicpm45v \
|
|
-t ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:minicpm45v \
|
|
-t ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:minicpm45v-gpu \
|
|
-t ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:latest \
|
|
.
|
|
|
|
# Build CPU variant
|
|
echo -e "${GREEN}Building MiniCPM-V 4.5 CPU variant...${NC}"
|
|
docker build \
|
|
-f Dockerfile_minicpm45v_cpu \
|
|
-t ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:minicpm45v-cpu \
|
|
.
|
|
|
|
# Build PaddleOCR GPU variant
|
|
echo -e "${GREEN}Building PaddleOCR GPU variant...${NC}"
|
|
docker build \
|
|
-f Dockerfile_paddleocr \
|
|
-t ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:paddleocr \
|
|
-t ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:paddleocr-gpu \
|
|
.
|
|
|
|
# Build PaddleOCR CPU variant
|
|
echo -e "${GREEN}Building PaddleOCR CPU variant...${NC}"
|
|
docker build \
|
|
-f Dockerfile_paddleocr_cpu \
|
|
-t ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:paddleocr-cpu \
|
|
.
|
|
|
|
echo -e "${GREEN}All images built successfully!${NC}"
|
|
echo ""
|
|
echo "Available images:"
|
|
echo " MiniCPM-V 4.5:"
|
|
echo " - ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:minicpm45v (GPU)"
|
|
echo " - ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:minicpm45v-cpu (CPU)"
|
|
echo " - ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:latest (GPU)"
|
|
echo ""
|
|
echo " PaddleOCR:"
|
|
echo " - ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:paddleocr (GPU)"
|
|
echo " - ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:paddleocr-gpu (GPU)"
|
|
echo " - ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:paddleocr-cpu (CPU)"
|