Sergey Penkovsky dc440a3938 test(gpt2): add unit tests for generation, cache behavior, and error conditions ...

- Covers forward pass with and without KV-cache
- Verifies correct sequence generation for greedy, top-k, and top-p sampling
- Adds ValueError test for exceeding max sequence length
- Uses small random toy config and minimal setup for fast test feedback

Motivation: Prevent regressions in decoding, sampling, and KV-cache logic in GPT2 implementation.

2025-10-15 14:36:32 +03:00

..

test_gpt2.py

test(gpt2): add unit tests for generation, cache behavior, and error conditions

2025-10-15 14:36:32 +03:00

test_gpt.py

refactor(core): refactor RoPE and MultiHeadAttention, add math-rich docs, expand tests, remove unused head_attention

2025-10-15 11:04:07 +03:00

test_mistral.py

test(mistral): add unit tests for model generation and cache

2025-10-15 13:20:50 +03:00