|
FROM s390x/python |
|
RUN wget https: |
|
&& bash Miniconda3-py311_23.5.2-0-Linux-s390x.sh -b \ |
|
&& rm -f Miniconda3-py311_23.5.2-0-Linux-s390x.sh |
|
RUN curl --proto '=https' --tlsv1.2 -sSf https: |
|
RUN /root/miniconda3/bin/conda install pytorch cpuonly -c pytorch -y |
|
WORKDIR /safetensors/ |
|
RUN /root/miniconda3/bin/pip install -U pip pytest |
|
# RUN /root/miniconda3/bin/pip install -U huggingface_hub |
|
# RUN /root/miniconda3/bin/python -c 'from huggingface_hub import hf_hub_download; filename = hf_hub_download("roberta-base", "model.safetensors")' |
|
COPY . . |
|
SHELL ["/bin/bash", "-c"] |
|
WORKDIR /safetensors/bindings/python/ |
|
RUN source /root/.cargo/env && /root/miniconda3/bin/pip install -e . |
|
RUN /root/miniconda3/bin/pytest -sv tests/test_pt_* tests/test_simple.py |
|
# RUN /root/miniconda3/bin/python -c 'from huggingface_hub import hf_hub_download; filename = hf_hub_download("roberta-base", "model.safetensors"); from safetensors.torch import load_file; weights = load_file(filename); assert weights["roberta.embeddings.position_embeddings.weight"][0][0].abs().item() > 1e-10' |
|
ENTRYPOINT /bin/bash |
|
|