FROM ghcr.io/ggml-org/llama.cpp:full

WORKDIR /app

# パッケージのインストール 
RUN apt update && apt install -y python3-pip
RUN pip install -U huggingface_hub

# モデルとmmprojの両方をHugging Faceからダウンロード 
RUN python3 -c 'from huggingface_hub import hf_hub_download; \
    repo="HauhauCS/Qwen3.5-4B-Uncensored-HauhauCS-Aggressive"; \
    hf_hub_download(repo_id=repo, filename="Qwen3.5-4B-Uncensored-HauhauCS-Aggressive-Q6_K.gguf", local_dir="/app"); \
    hf_hub_download(repo_id=repo, filename="mmproj-Qwen3.5-4B-Uncensored-HauhauCS-Aggressive-BF16.gguf", local_dir="/app")'

# サーバーの設定（mmprojを追加） 
CMD ["--server", \
     "-m", "/app/Qwen3.5-4B-Uncensored-HauhauCS-Aggressive-Q6_K.gguf", \
     "--mmproj", "/app/mmproj-Qwen3.5-4B-Uncensored-HauhauCS-Aggressive-BF16.gguf", \
     "--host", "0.0.0.0", \
     "--port", "7860", \
     "-t", "2", \
     "--cache-type-k", "q8_0", \
     "--cache-type-v", "iq4_nl", \
     "-c", "128000", \
     "-n", "38912"]