# Giai đoạn 1: Download model bằng Python
FROM python:3.10-slim AS downloader

# Cài đặt thư viện cần thiết
RUN pip install --no-cache-dir huggingface_hub

# Tải model (Thay 'id/model' bằng model bạn cần, ví dụ 'Qwen/Qwen2.5-7B-Instruct')
# local_dir='/model_weight' là nơi chứa file sau khi tải
RUN python -c "from huggingface_hub import snapshot_download; \
    snapshot_download(repo_id='Qwen/Qwen2.5-7B-Instruct', \
    local_dir='/model_weight', \
    ignore_patterns=['*.msgpack', '*.h5', '*.ot'])"

# Giai đoạn 2: Tạo Image siêu nhẹ chỉ chứa Weight
# Sử dụng alpine hoặc busybox để image có sẵn shell (dễ cp ra ngoài)
FROM alpine:latest
COPY --from=downloader /model_weight /models
CMD ["sh", "-c", "sleep infinity"]