Go to file
2023-02-24 16:29:36 -08:00
cacheflow Fix a bug in tying OPT embeddings (#1) 2023-02-24 16:29:36 -08:00
csrc Add reshape_and_cache op 2023-02-18 19:22:57 +00:00
tests Add tests for kernels 2023-02-18 19:23:07 +00:00
.gitignore Add gitignore 2023-02-16 07:47:21 +00:00
README.md Add README 2023-02-24 12:04:49 +00:00
server.py Clean up the server script 2023-02-24 11:56:21 +00:00
setup.py cache_kernel -> cache_kernels 2023-02-16 20:05:45 +00:00

CacheFlow

Installation

pip install cmake torch transformers
pip install -e .

Run

python server.py