| .. |
|
__init__.py
|
07aa2a492f
upstream: add option to specify tokenizer
|
1 жил өмнө |
|
block.py
|
7df7b8ca53
optimization: reduce end-to-end overhead from python obj allocation (#666)
|
6 сар өмнө |
|
config.py
|
92cee435e2
rocm: add more quants, fix _scaled_mm call (#1062)
|
2 сар өмнө |
|
connections.py
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
6 сар өмнө |
|
envs.py
|
9a7d5514c4
feat: introduce MQAphroditeEngine (#1056)
|
2 сар өмнө |
|
grammar.py
|
8a71788372
Add OLMoE (#772)
|
4 сар өмнө |
|
logger.py
|
22a4cd4595
core: fix spec decode metrics and envs circular import (#889)
|
2 сар өмнө |
|
logits_processor.py
|
8a71788372
Add OLMoE (#772)
|
4 сар өмнө |
|
outputs.py
|
055c8905a3
api: add sampling/engine option to return only deltas or final output (#1035)
|
2 сар өмнө |
|
pooling_params.py
|
2f61644f6e
SPMD optimizations (#824)
|
3 сар өмнө |
|
sampling_params.py
|
814c850d89
fix: validate `n` in the sampling params (#1075)
|
2 сар өмнө |
|
sequence.py
|
651678d2df
VLM: use `SequenceData.from_token_counts` to create dummy data (#1093)
|
1 сар өмнө |
|
test_utils.py
|
8a71788372
Add OLMoE (#772)
|
4 сар өмнө |
|
utils.py
|
d44da0332c
misc: rename `CudaMemoryProfiler` to `DeviceMemoryProfiler` (#1099)
|
1 сар өмнө |
AlpinDale
d44da0332c
misc: rename `CudaMemoryProfiler` to `DeviceMemoryProfiler` (#1099)