 AlpinDale
|
6dfca14e1f
compute logprobs with log_softmax instead of log
|
vor 1 Jahr |
|
AlpinDale
|
7a85354b69
add logits back
|
vor 1 Jahr |
|
AlpinDale
|
45f6d9f923
initial refactor commit
|
vor 1 Jahr |
|
AlpinDale
|
23389d0108
zero out a variable instead of vector in kernels
|
vor 1 Jahr |
|
AlpinDale
|
aba8b0b17a
add rope theta support and bump transformers
|
vor 1 Jahr |
|
AlpinDale
|
62dfd1a883
fix top_k implementation
|
vor 1 Jahr |
 AlpinDale
|
76b2e4a445
Merge dev branch into main (#7)
|
vor 1 Jahr |
|
AlpinDale
|
76162fe366
chore: optimize sampler for gpt-j
|
vor 1 Jahr |
|
AlpinDale
|
44faffe272
chore: remove comparison of floats with 0
|
vor 1 Jahr |
|
AlpinDale
|
48a75478cb
feat: add generation sampler
|
vor 1 Jahr |
