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