├── .gitignore ├── README.md ├── assets ├── loss.png ├── menu.png ├── perfect_solution_rate.png └── token_usage.png ├── demo.py ├── formatter.py ├── generate_sudoku_data.py ├── minimum_inference.py ├── model.py ├── rwkv_model.py ├── sudoku_data.jsonl ├── sudoku_rwkv_20241120.pth ├── sudoku_vocab.txt └── utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | /.idea 2 | /__pycache__ 3 | /.history -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jellyfish042/Sudoku-RWKV/HEAD/README.md -------------------------------------------------------------------------------- /assets/loss.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jellyfish042/Sudoku-RWKV/HEAD/assets/loss.png -------------------------------------------------------------------------------- /assets/menu.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jellyfish042/Sudoku-RWKV/HEAD/assets/menu.png -------------------------------------------------------------------------------- /assets/perfect_solution_rate.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jellyfish042/Sudoku-RWKV/HEAD/assets/perfect_solution_rate.png -------------------------------------------------------------------------------- /assets/token_usage.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jellyfish042/Sudoku-RWKV/HEAD/assets/token_usage.png -------------------------------------------------------------------------------- /demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jellyfish042/Sudoku-RWKV/HEAD/demo.py -------------------------------------------------------------------------------- /formatter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jellyfish042/Sudoku-RWKV/HEAD/formatter.py -------------------------------------------------------------------------------- /generate_sudoku_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jellyfish042/Sudoku-RWKV/HEAD/generate_sudoku_data.py -------------------------------------------------------------------------------- /minimum_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jellyfish042/Sudoku-RWKV/HEAD/minimum_inference.py -------------------------------------------------------------------------------- /model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jellyfish042/Sudoku-RWKV/HEAD/model.py -------------------------------------------------------------------------------- /rwkv_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jellyfish042/Sudoku-RWKV/HEAD/rwkv_model.py -------------------------------------------------------------------------------- /sudoku_data.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jellyfish042/Sudoku-RWKV/HEAD/sudoku_data.jsonl -------------------------------------------------------------------------------- /sudoku_rwkv_20241120.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jellyfish042/Sudoku-RWKV/HEAD/sudoku_rwkv_20241120.pth -------------------------------------------------------------------------------- /sudoku_vocab.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jellyfish042/Sudoku-RWKV/HEAD/sudoku_vocab.txt -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jellyfish042/Sudoku-RWKV/HEAD/utils.py --------------------------------------------------------------------------------