books/LLMs-from-scratch
1
0
Fork 0
mirror of https://github.com/rasbt/LLMs-from-scratch.git synced 2026-04-10 12:33:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix bug in masking when kv cache is used. (#697)

Browse Source 
* Fix bug in masking when kv cache is used.

* add tests

* dd tests

* upd

* add kv cache test to gh workflow

* explicit mask slicing

* upd

---------

Co-authored-by: rasbt <mail@sebastianraschka.com>
This commit is contained in:
Martin Ma
2025-06-23 11:12:56 -07:00
committed by GitHub
parent 37b26c2e04
commit 6522be94be
5 changed files with 179 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
.github/workflows/basic-tests-linux-uv.yml vendored
View File

@@ -49,6 +49,7 @@ jobs:
source .venv/bin/activate
pytest --ruff setup/02_installing-python-libraries/tests.py
pytest --ruff ch04/01_main-chapter-code/tests.py
pytest --ruff ch04/03_kv-cache/tests.py
pytest --ruff ch05/01_main-chapter-code/tests.py
pytest --ruff ch05/07_gpt_to_llama/tests/tests.py
pytest --ruff ch06/01_main-chapter-code/tests.py