.. |
__init__.py
|
9d81716bfd
[v0.5.3] Release Candidate (#388)
|
8 meses atrás |
batch_expansion.py
|
16b587c104
fix: hidden states handling in batch expansion for spec decoding (#839)
|
1 mês atrás |
draft_model_runner.py
|
1405051912
attention: add `AttentionState` abstraction (#863)
|
1 mês atrás |
interfaces.py
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
4 meses atrás |
medusa_worker.py
|
22a4cd4595
core: fix spec decode metrics and envs circular import (#889)
|
3 semanas atrás |
metrics.py
|
2f61644f6e
SPMD optimizations (#824)
|
1 mês atrás |
mlp_speculator_worker.py
|
22a4cd4595
core: fix spec decode metrics and envs circular import (#889)
|
3 semanas atrás |
multi_step_worker.py
|
1394008421
chore: decouple `should_modify_greedy_probs_inplace (#671)
|
4 meses atrás |
ngram_worker.py
|
2f61644f6e
SPMD optimizations (#824)
|
1 mês atrás |
proposer_worker_base.py
|
2f61644f6e
SPMD optimizations (#824)
|
1 mês atrás |
smaller_tp_proposer_worker.py
|
2f61644f6e
SPMD optimizations (#824)
|
1 mês atrás |
spec_decode_worker.py
|
16b587c104
fix: hidden states handling in batch expansion for spec decoding (#839)
|
1 mês atrás |
target_model_runner.py
|
89a2c6dee1
chore: refactor `MultiModalConfig` initialization and profiling (#745)
|
3 meses atrás |
top1_proposer.py
|
16b587c104
fix: hidden states handling in batch expansion for spec decoding (#839)
|
1 mês atrás |
util.py
|
16b587c104
fix: hidden states handling in batch expansion for spec decoding (#839)
|
1 mês atrás |