fix: Dockerfile.build missing cmake in rwkv example (#301)

Signed-off-by: Tyler Gillson <tyler.gillson@gmail.com>
swagger2
Tyler Gillson 1 year ago committed by GitHub
parent fc59f74849
commit 207ce81e4a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 2
      examples/rwkv/.gitignore
  2. 2
      examples/rwkv/Dockerfile.build

@ -0,0 +1,2 @@
models/rwkv
models/rwkv.tokenizer.json

@ -1,5 +1,7 @@
FROM python FROM python
RUN apt-get update && apt-get -y install cmake
# convert the model (one-off) # convert the model (one-off)
RUN pip3 install torch numpy RUN pip3 install torch numpy

Loading…
Cancel
Save