HuggingFaceModelDownloader/Dockerfile at master · bodaay/HuggingFaceModelDownloader · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
# =============================================================================
# HuggingFace Model Downloader - Docker Image (v3)
# =============================================================================
# Multi-stage build for minimal image size
#
# Build:
#   docker build -t hfdownloader .
#
# Run CLI (v3 - uses HuggingFace cache structure):
#   docker run --rm -v ~/.cache/huggingface:/home/hfdownloader/.cache/huggingface \
#     hfdownloader download TheBloke/Mistral-7B-Instruct-v0.2-GGUF
#
# Run CLI with filter:
#   docker run --rm -v ~/.cache/huggingface:/home/hfdownloader/.cache/huggingface \
#     hfdownloader download TheBloke/Mistral-7B-Instruct-v0.2-GGUF:q4_k_m
#
# Run Web Server:
#   docker run --rm -p 8080:8080 \
#     -v ~/.cache/huggingface:/home/hfdownloader/.cache/huggingface \
#     hfdownloader serve --port 8080
#
# With HuggingFace token (for private/gated models):
#   docker run --rm -e HF_TOKEN=hf_xxx -p 8080:8080 \
#     -v ~/.cache/huggingface:/home/hfdownloader/.cache/huggingface \
#     hfdownloader serve
#
# Legacy mode (v2 behavior - flat directory structure):
#   docker run --rm -v ./models:/data hfdownloader download TheBloke/Mistral-7B-Instruct-v0.2-GGUF --legacy -o /data
#
# Credits: Original Docker support suggested by cdeving (#50)
# =============================================================================

# Build stage
FROM golang:1.24-alpine AS builder

# Install build dependencies
RUN apk add --no-cache git ca-certificates

WORKDIR /src

# Cache dependencies
COPY go.mod go.sum ./
RUN go mod download

# Copy source code
COPY . .

# Build the binary
RUN CGO_ENABLED=0 GOOS=linux go build -ldflags="-s -w" -o /hfdownloader ./cmd/hfdownloader

# =============================================================================
# Final stage - minimal image
# =============================================================================
FROM alpine:3.19

# Install ca-certificates for HTTPS
RUN apk add --no-cache ca-certificates tzdata

# Create non-root user
RUN adduser -D -u 1000 hfdownloader

# Copy binary from builder
COPY --from=builder /hfdownloader /usr/local/bin/hfdownloader

# Create HuggingFace cache directory (v3 default) and legacy data directory
RUN mkdir -p /home/hfdownloader/.cache/huggingface/hub \
             /home/hfdownloader/.cache/huggingface/models \
             /home/hfdownloader/.cache/huggingface/datasets \
             /data/Models /data/Datasets && \
    chown -R hfdownloader:hfdownloader /home/hfdownloader /data

# Switch to non-root user
USER hfdownloader

# Set HF_HOME for the container
ENV HF_HOME=/home/hfdownloader/.cache/huggingface

WORKDIR /home/hfdownloader

# Default to showing help
ENTRYPOINT ["/usr/local/bin/hfdownloader"]
CMD ["--help"]

# Expose web server port
EXPOSE 8080

# Labels
LABEL org.opencontainers.image.source="https://github.com/bodaay/HuggingFaceModelDownloader"
LABEL org.opencontainers.image.description="Fast, concurrent CLI and Web UI for downloading models and datasets from HuggingFace"
LABEL org.opencontainers.image.licenses="Apache-2.0"