rwkv fixes and examples (#185)

This commit is contained in:
Ettore Di Giacinto 2023-05-04 17:32:23 +02:00 committed by GitHub
parent 6352308882
commit fdf75c6d0e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
9 changed files with 125 additions and 1 deletions

View file

@ -0,0 +1,10 @@
FROM python
# convert the model (one-off)
RUN pip3 install torch numpy
WORKDIR /build
COPY ./scripts/ .
RUN git clone --recurse-submodules https://github.com/saharNooby/rwkv.cpp && cd rwkv.cpp && cmake . && cmake --build . --config Release
ENTRYPOINT [ "/build/build.sh" ]