| .. |
|
fused_moe
|
81c28d2a7f
fix: use nvml to get consistent device names (#739)
|
3 mēneši atpakaļ |
|
mamba
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
4 mēneši atpakaļ |
|
ops
|
8a71788372
Add OLMoE (#772)
|
2 mēneši atpakaļ |
|
__init__.py
|
07aa2a492f
upstream: add option to specify tokenizer
|
1 gadu atpakaļ |
|
activation.py
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
4 mēneši atpakaļ |
|
layernorm.py
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
4 mēneši atpakaļ |
|
linear.py
|
f98e7b2f8c
feat: add HQQ quantization support (#795)
|
2 mēneši atpakaļ |
|
logits_processor.py
|
0e558e9b2f
fix: loading chameleon model with TP>1 (#695)
|
4 mēneši atpakaļ |
|
pooler.py
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
4 mēneši atpakaļ |
|
rejection_sampler.py
|
e3a53712f2
fix: mlpspeculator with padded vocab (#669)
|
4 mēneši atpakaļ |
|
rotary_embedding.py
|
4434c4db84
chore: refactor llama3 rope (#748)
|
3 mēneši atpakaļ |
|
sampler.py
|
afb710e3e4
dry: use gpu for z_array computation
|
1 mēnesi atpakaļ |
|
spec_decode_base_sampler.py
|
09b82f9963
feat: Add support for GPU device selection in SpecDecodeBaseSampler (#629)
|
4 mēneši atpakaļ |
|
typical_acceptance_sampler.py
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
4 mēneši atpakaļ |
|
vocab_parallel_embedding.py
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
4 mēneši atpakaļ |
AlpinDale
afb710e3e4
dry: use gpu for z_array computation