From 207ce81e4af83db28ad09f1e2f68ef8304e0c628 Mon Sep 17 00:00:00 2001 From: Tyler Gillson Date: Thu, 18 May 2023 16:08:20 -0700 Subject: [PATCH] fix: Dockerfile.build missing cmake in rwkv example (#301) Signed-off-by: Tyler Gillson --- examples/rwkv/.gitignore | 2 ++ examples/rwkv/Dockerfile.build | 2 ++ 2 files changed, 4 insertions(+) create mode 100644 examples/rwkv/.gitignore diff --git a/examples/rwkv/.gitignore b/examples/rwkv/.gitignore new file mode 100644 index 0000000..ab3629c --- /dev/null +++ b/examples/rwkv/.gitignore @@ -0,0 +1,2 @@ +models/rwkv +models/rwkv.tokenizer.json \ No newline at end of file diff --git a/examples/rwkv/Dockerfile.build b/examples/rwkv/Dockerfile.build index c62024d..491f9cc 100644 --- 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