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

Signed-off-by: Tyler Gillson <tyler.gillson@gmail.com>
2 years ago · 207ce81e4a
parent fc59f74849
commit 207ce81e4a
2 changed files with 4 additions and 0 deletions
--- a/examples/rwkv/.gitignore
+++ b/examples/rwkv/.gitignore
@ -0,0 +1,2 @@
+models/rwkv
+models/rwkv.tokenizer.json
--- a/examples/rwkv/Dockerfile.build
+++ b/examples/rwkv/Dockerfile.build
@ -1,5 +1,7 @@
 FROM python

+RUN apt-get update && apt-get -y install cmake
+
 # convert the model (one-off)
 RUN pip3 install torch numpy