.. |
__init__.py
|
04b53d2db5
chore: add initializer files
|
vor 1 Jahr |
cache_engine.py
|
a1f18f17e6
modify the cache engine and model runner/worker to support mamba states
|
vor 8 Monaten |
cpu_model_runner.py
|
6e0761ba5d
make init_distributed_environment compatible with init_process_group
|
vor 9 Monaten |
cpu_worker.py
|
083ba7b452
roll back chunked prefill changes to SDPA, isolate cpu worker
|
vor 9 Monaten |
model_runner.py
|
a1f18f17e6
modify the cache engine and model runner/worker to support mamba states
|
vor 8 Monaten |
neuron_model_runner.py
|
0f1399c135
feat: attention refactor part 2
|
vor 9 Monaten |
neuron_worker.py
|
4d33ce60da
feat: Triton flash attention backend for ROCm (#407)
|
vor 9 Monaten |
worker.py
|
a1f18f17e6
modify the cache engine and model runner/worker to support mamba states
|
vor 8 Monaten |
worker_base.py
|
8c67b37131
fix docstrings
|
vor 9 Monaten |