File size: 277 Bytes
2935c37 41b30a0 1a61dc5 |
1 2 3 4 5 |
FROM nvidia/cuda:12.4.0-runtime-ubuntu20.04
RUN apt-get update && apt-get install -y python3 python3-pip wget
RUN wget --header="Authorization: Bearer ${HF_TOKEN}" https://huggingface.co/nvidia/Hymba-1.5B-Base/resolve/main/setup.sh
RUN bash setup.sh
CMD [ "python", "chat.py" ] |