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

Signed-off-by: Tyler Gillson <tyler.gillson@gmail.com>
This commit is contained in:
Tyler Gillson authored and GitHub committed 2023-05-19 01:08:20 +02:00
1 parent fc59f74849
commit 207ce81e4a
2 files changed
+4

No files matched your search

+2
View File
@@ -0,0 +1,2 @@
models/rwkv
models/rwkv.tokenizer.json
+2
View File
@@ -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