.. |
__init__.py
|
07aa2a492f
upstream: add option to specify tokenizer
|
há 1 ano atrás |
activation.py
|
32844c1522
add GELU kernels and remove compile bloat
|
há 1 ano atrás |
attention.py
|
e77960c57e
use float datatype for RoPE
|
há 1 ano atrás |
layernorm.py
|
f2fe14ec1c
fix: epsilon variance value for RMSNorm
|
há 1 ano atrás |
sampler.py
|
6dfca14e1f
compute logprobs with log_softmax instead of log
|
há 1 ano atrás |