├── .cargo └── config.toml ├── .gitignore ├── Cargo.lock ├── Cargo.toml ├── README.md ├── dataset.txt ├── examples ├── convert.rs ├── quantize.rs └── rwkv.rs └── src ├── lib.rs └── models ├── mod.rs ├── quantized ├── mod.rs ├── rwkv5.rs └── rwkv6.rs ├── rwkv5.rs └── rwkv6.rs /.cargo/config.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nkypy/candle-rwkv/HEAD/.cargo/config.toml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nkypy/candle-rwkv/HEAD/.gitignore -------------------------------------------------------------------------------- /Cargo.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nkypy/candle-rwkv/HEAD/Cargo.lock -------------------------------------------------------------------------------- /Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nkypy/candle-rwkv/HEAD/Cargo.toml -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nkypy/candle-rwkv/HEAD/README.md -------------------------------------------------------------------------------- /dataset.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nkypy/candle-rwkv/HEAD/dataset.txt -------------------------------------------------------------------------------- /examples/convert.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nkypy/candle-rwkv/HEAD/examples/convert.rs -------------------------------------------------------------------------------- /examples/quantize.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nkypy/candle-rwkv/HEAD/examples/quantize.rs -------------------------------------------------------------------------------- /examples/rwkv.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nkypy/candle-rwkv/HEAD/examples/rwkv.rs -------------------------------------------------------------------------------- /src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nkypy/candle-rwkv/HEAD/src/lib.rs -------------------------------------------------------------------------------- /src/models/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nkypy/candle-rwkv/HEAD/src/models/mod.rs -------------------------------------------------------------------------------- /src/models/quantized/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nkypy/candle-rwkv/HEAD/src/models/quantized/mod.rs -------------------------------------------------------------------------------- /src/models/quantized/rwkv5.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nkypy/candle-rwkv/HEAD/src/models/quantized/rwkv5.rs -------------------------------------------------------------------------------- /src/models/quantized/rwkv6.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nkypy/candle-rwkv/HEAD/src/models/quantized/rwkv6.rs -------------------------------------------------------------------------------- /src/models/rwkv5.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nkypy/candle-rwkv/HEAD/src/models/rwkv5.rs -------------------------------------------------------------------------------- /src/models/rwkv6.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nkypy/candle-rwkv/HEAD/src/models/rwkv6.rs --------------------------------------------------------------------------------