| .. |
|
__init__.py
|
07aa2a492f
upstream: add option to specify tokenizer
|
1 year ago |
|
block.py
|
7df7b8ca53
optimization: reduce end-to-end overhead from python obj allocation (#666)
|
4 months ago |
|
config.py
|
92cee435e2
rocm: add more quants, fix _scaled_mm call (#1062)
|
6 days ago |
|
connections.py
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
4 months ago |
|
envs.py
|
9a7d5514c4
feat: introduce MQAphroditeEngine (#1056)
|
1 week ago |
|
grammar.py
|
8a71788372
Add OLMoE (#772)
|
2 months ago |
|
logger.py
|
22a4cd4595
core: fix spec decode metrics and envs circular import (#889)
|
3 weeks ago |
|
logits_processor.py
|
8a71788372
Add OLMoE (#772)
|
2 months ago |
|
outputs.py
|
055c8905a3
api: add sampling/engine option to return only deltas or final output (#1035)
|
1 week ago |
|
pooling_params.py
|
2f61644f6e
SPMD optimizations (#824)
|
1 month ago |
|
sampling_params.py
|
814c850d89
fix: validate `n` in the sampling params (#1075)
|
2 days ago |
|
sequence.py
|
055c8905a3
api: add sampling/engine option to return only deltas or final output (#1035)
|
1 week ago |
|
test_utils.py
|
8a71788372
Add OLMoE (#772)
|
2 months ago |
|
utils.py
|
9bdf8d5bfa
mamba: enable continuous batching for mamba kernels (#1055)
|
1 week ago |
AlpinDale
814c850d89
fix: validate `n` in the sampling params (#1075)