]> git.djapps.eu Git - pkg/ggml/sources/ggml/commit
llama: add support for QRWKV6 model architecture (llama/11001)
authorMolly Sophia <redacted>
Fri, 10 Jan 2025 01:58:08 +0000 (09:58 +0800)
committerGeorgi Gerganov <redacted>
Tue, 14 Jan 2025 07:36:36 +0000 (09:36 +0200)
commit894e5827c50385001a8695ecc20f4ed22f2e84cb
tree470c0da5e1187f9cde4401a362c1d022f425cf12
parent17eb09dde1d1a87ea1d091c6592e902a501ff886
llama: add support for QRWKV6 model architecture (llama/11001)

llama: add support for QRWKV6 model architecture (llama/11001)

* WIP: Add support for RWKV6Qwen2

Signed-off-by: Molly Sophia <redacted>
* RWKV: Some graph simplification

Signed-off-by: Molly Sophia <redacted>
* Add support for RWKV6Qwen2 with cpu and cuda GLA

Signed-off-by: Molly Sophia <redacted>
* RWKV6[QWEN2]: Concat lerp weights together to reduce cpu overhead

Signed-off-by: Molly Sophia <redacted>
* Fix some typos

Signed-off-by: Molly Sophia <redacted>
* code format changes

Signed-off-by: Molly Sophia <redacted>
* Fix wkv test & add gla test

Signed-off-by: Molly Sophia <redacted>
* Fix cuda warning

Signed-off-by: Molly Sophia <redacted>
* Update README.md

Signed-off-by: Molly Sophia <redacted>
* Update ggml/src/ggml-cuda/gla.cu

Co-authored-by: Georgi Gerganov <redacted>
* Fix fused lerp weights loading with RWKV6

Signed-off-by: Molly Sophia <redacted>
* better sanity check skipping for QRWKV6 in llama-quant

thanks @compilade

Signed-off-by: Molly Sophia <redacted>
Co-authored-by: compilade <redacted>
---------

Signed-off-by: Molly Sophia <redacted>
Co-authored-by: Georgi Gerganov <redacted>
Co-authored-by: compilade <redacted>
include/ggml.h
src/ggml-cpu/ggml-cpu.c
src/ggml-cuda/ggml-cuda.cu
src/ggml-cuda/gla.cu [new file with mode: 0644]
src/ggml-cuda/gla.cuh [new file with mode: 0644]
src/ggml-cuda/wkv6.cu
src/ggml-sycl/wkv6.cpp
src/ggml-vulkan/ggml-vulkan.cpp
src/ggml.c
tests/test-backend-ops.cpp