r4/Dockerfile.r4userembeddingwithmodel at main · beclab/r4 · 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
FROM ubuntu:jammy As builder

# RUN apk add openssl-dev musl-dev g++

RUN mkdir -p /userembedding/model && \
    apt update && \
    apt install curl -y && \
    curl https://sh.rustup.rs -sSf | bash -s -- -y && \
    # echo 'source $HOME/.cargo/env' >> $HOME/.bashrc && \
    apt install build-essential -y && \
    apt-get install libssl-dev -y && \
    apt-get install pkg-config -y

ENV PATH="/root/.cargo/bin:${PATH}"
WORKDIR /userembedding
COPY user-embedding/src  /userembedding/src
COPY user-embedding/Cargo.toml /userembedding/Cargo.toml
RUN cargo build --release
# RUN  /userembedding/target/release/downloadmodel  # temp removed


FROM  ubuntu:jammy

# Import from builder.
ENV MODEL_INTERNET="false"


WORKDIR /userembedding
RUN mkdir -p /userembedding/model

# Copy our build
COPY --from=builder /userembedding/model /userembedding/model
COPY --from=builder /userembedding/target/release/userembedding ./
COPY --from=builder /userembedding/target/release/downloadmodel ./
COPY --from=builder /userembedding/target/release/buildmodelfromlocal ./


CMD ["/userembedding/userembedding"]
# ENTRYPOINT ["tail", "-f", "/dev/null"]