diff --git a/.github/PULL_REQUEST_TEMPLATE.md b/.github/PULL_REQUEST_TEMPLATE.md index d3f0e0f0..0703b15b 100644 --- a/.github/PULL_REQUEST_TEMPLATE.md +++ b/.github/PULL_REQUEST_TEMPLATE.md @@ -2,9 +2,13 @@ Description of your PR. Fixes # (issue) (if applicable) -## New press checklist (if applicable) +## Checklist -- [ ] I added `mypress_press.py` in the `presses` directory -- [ ] I added `MyPress` in `__init__.py` -- [ ] I updated the `README.md` with a 1 liner about my new press in the Available presses section -- [ ] I added my press in the `default_presses` list in `tests/default_presses.py` +- Tests are working (make test) +- Code is formatted correctly (make style, on errors try fix with make format) +- Copyright header is included +- [ ] All commits are signed-off using `git commit -s` +- [ ] (new press) `mypress_press.py` is in the `presses` directory +- [ ] (new press) `MyPress` is in `__init__.py` +- [ ] (new press) `README.md` is updated with a 1 liner about the new press in the Available presses section +- [ ] (new press) new press is in the `default_presses` list in `tests/default_presses.py` diff --git a/CITATION.cff b/CITATION.cff new file mode 100644 index 00000000..c61ea0ae --- /dev/null +++ b/CITATION.cff @@ -0,0 +1,13 @@ +cff-version: 1.2.0 +message: "If you use kvpress, please cite it as below." +authors: +- family-names: "Jegou" + given-names: "Simon" +- family-names: "Jeblick" + given-names: "Maximilian" +- family-names: "Austin" + given-names: "David" +title: "kvpress" +date-released: 2024-11-13 +year: 2024 +url: "https://github.com/NVIDIA/kvpress" \ No newline at end of file diff --git a/README.md b/README.md index ae5a00af..53514785 100644 --- a/README.md +++ b/README.md @@ -113,7 +113,6 @@ By default, the `DynamicCache` is used (no quantization). > [!IMPORTANT] > To use the `QuantizedCache`, you need to install additional dependencies (_e.g._ `pip install optimum-quanto`). - ## FAQ
@@ -124,6 +123,19 @@ By default, the `DynamicCache` is used (no quantization). Some presses depend on the model architecture (_e.g._ `ExpectedAttentionPress` or `SnapKVPress`) hence they might not work with all models. We tested support for `LlamaForCausalLM`, `MistralForCausalLM`, `Phi3ForCausalLM` and `Qwen2ForCausalLM` but many other models might be supported out of the box because their implementation is often similar in transformers.
+
+ +### How to run inference on multiple GPUs ? + + +kvpress supports multi-GPU inference through [accelerate](https://huggingface.co/docs/accelerate/en/index): + +```python +pipe = pipeline("kv-press-text-generation", model=model, device_map="auto") +``` + +
+
diff --git a/evaluation/__init__.py b/evaluation/__init__.py index 299764fc..9adfdfcd 100644 --- a/evaluation/__init__.py +++ b/evaluation/__init__.py @@ -1,2 +1,2 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/evaluation/evaluate.py b/evaluation/evaluate.py index d06b5668..7ee65d59 100644 --- a/evaluation/evaluate.py +++ b/evaluation/evaluate.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 import json diff --git a/evaluation/infinite_bench/calculate_metrics.py b/evaluation/infinite_bench/calculate_metrics.py index 8b32549a..e9937896 100644 --- a/evaluation/infinite_bench/calculate_metrics.py +++ b/evaluation/infinite_bench/calculate_metrics.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/evaluation/infinite_bench/create_huggingface_dataset.py b/evaluation/infinite_bench/create_huggingface_dataset.py index 8fdbf466..60303fa5 100644 --- a/evaluation/infinite_bench/create_huggingface_dataset.py +++ b/evaluation/infinite_bench/create_huggingface_dataset.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/evaluation/loogle/__init__.py b/evaluation/loogle/__init__.py index 299764fc..9adfdfcd 100644 --- a/evaluation/loogle/__init__.py +++ b/evaluation/loogle/__init__.py @@ -1,2 +1,2 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/evaluation/loogle/calculate_metrics.py b/evaluation/loogle/calculate_metrics.py index ec43c9c7..7deb02bd 100644 --- a/evaluation/loogle/calculate_metrics.py +++ b/evaluation/loogle/calculate_metrics.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/evaluation/loogle/create_huggingface_dataset.py b/evaluation/loogle/create_huggingface_dataset.py index 1527931c..ef2fd15b 100644 --- a/evaluation/loogle/create_huggingface_dataset.py +++ b/evaluation/loogle/create_huggingface_dataset.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/evaluation/ruler/__init__.py b/evaluation/ruler/__init__.py index 299764fc..9adfdfcd 100644 --- a/evaluation/ruler/__init__.py +++ b/evaluation/ruler/__init__.py @@ -1,2 +1,2 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/evaluation/ruler/calculate_metrics.py b/evaluation/ruler/calculate_metrics.py index 86be5ab5..16d1b924 100644 --- a/evaluation/ruler/calculate_metrics.py +++ b/evaluation/ruler/calculate_metrics.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/evaluation/ruler/create_huggingface_dataset.py b/evaluation/ruler/create_huggingface_dataset.py index ecfbaf0d..b31f14bf 100644 --- a/evaluation/ruler/create_huggingface_dataset.py +++ b/evaluation/ruler/create_huggingface_dataset.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/evaluation/zero_scrolls/__init__.py b/evaluation/zero_scrolls/__init__.py index 299764fc..9adfdfcd 100644 --- a/evaluation/zero_scrolls/__init__.py +++ b/evaluation/zero_scrolls/__init__.py @@ -1,2 +1,2 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/evaluation/zero_scrolls/calculate_metrics.py b/evaluation/zero_scrolls/calculate_metrics.py index b7afa052..d34c22e1 100644 --- a/evaluation/zero_scrolls/calculate_metrics.py +++ b/evaluation/zero_scrolls/calculate_metrics.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/evaluation/zero_scrolls/create_huggingface_dataset.py b/evaluation/zero_scrolls/create_huggingface_dataset.py index 3a313a3d..7e3e9d45 100644 --- a/evaluation/zero_scrolls/create_huggingface_dataset.py +++ b/evaluation/zero_scrolls/create_huggingface_dataset.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/kvpress/__init__.py b/kvpress/__init__.py index ed1bb151..f53261bc 100644 --- a/kvpress/__init__.py +++ b/kvpress/__init__.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/kvpress/pipeline.py b/kvpress/pipeline.py index c9a0f1a7..88708495 100644 --- a/kvpress/pipeline.py +++ b/kvpress/pipeline.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/kvpress/presses/__init__.py b/kvpress/presses/__init__.py index 299764fc..9adfdfcd 100644 --- a/kvpress/presses/__init__.py +++ b/kvpress/presses/__init__.py @@ -1,2 +1,2 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/kvpress/presses/adakv_press.py b/kvpress/presses/adakv_press.py index cbb7e302..75d85a2f 100644 --- a/kvpress/presses/adakv_press.py +++ b/kvpress/presses/adakv_press.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/kvpress/presses/base_press.py b/kvpress/presses/base_press.py index 88a2d025..187c8b13 100644 --- a/kvpress/presses/base_press.py +++ b/kvpress/presses/base_press.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/kvpress/presses/chunk_press.py b/kvpress/presses/chunk_press.py index 2798a144..2d082c66 100644 --- a/kvpress/presses/chunk_press.py +++ b/kvpress/presses/chunk_press.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 from dataclasses import dataclass diff --git a/kvpress/presses/chunkkv_press.py b/kvpress/presses/chunkkv_press.py index 54d56cf5..e540d12a 100644 --- a/kvpress/presses/chunkkv_press.py +++ b/kvpress/presses/chunkkv_press.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 from dataclasses import dataclass diff --git a/kvpress/presses/criticalkv_press.py b/kvpress/presses/criticalkv_press.py index 8a9042e5..a95a22ec 100644 --- a/kvpress/presses/criticalkv_press.py +++ b/kvpress/presses/criticalkv_press.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 import logging diff --git a/kvpress/presses/duo_attention_press.py b/kvpress/presses/duo_attention_press.py index c48c2e27..e8966c6c 100644 --- a/kvpress/presses/duo_attention_press.py +++ b/kvpress/presses/duo_attention_press.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 from contextlib import contextmanager diff --git a/kvpress/presses/expected_attention_press.py b/kvpress/presses/expected_attention_press.py index 74b2a290..7dabc910 100644 --- a/kvpress/presses/expected_attention_press.py +++ b/kvpress/presses/expected_attention_press.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/kvpress/presses/key_rerotation_press.py b/kvpress/presses/key_rerotation_press.py index 0675fbcb..f4312b72 100644 --- a/kvpress/presses/key_rerotation_press.py +++ b/kvpress/presses/key_rerotation_press.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/kvpress/presses/knorm_press.py b/kvpress/presses/knorm_press.py index 34b695bc..763ee5f4 100644 --- a/kvpress/presses/knorm_press.py +++ b/kvpress/presses/knorm_press.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/kvpress/presses/observed_attention_press.py b/kvpress/presses/observed_attention_press.py index 4e9e78c9..ce4416fa 100644 --- a/kvpress/presses/observed_attention_press.py +++ b/kvpress/presses/observed_attention_press.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/kvpress/presses/per_layer_compression_press.py b/kvpress/presses/per_layer_compression_press.py index 31e87d61..da0d86f9 100644 --- a/kvpress/presses/per_layer_compression_press.py +++ b/kvpress/presses/per_layer_compression_press.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/kvpress/presses/random_press.py b/kvpress/presses/random_press.py index 063e8082..52120402 100644 --- a/kvpress/presses/random_press.py +++ b/kvpress/presses/random_press.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/kvpress/presses/scorer_press.py b/kvpress/presses/scorer_press.py index ea97eab4..475728f5 100644 --- a/kvpress/presses/scorer_press.py +++ b/kvpress/presses/scorer_press.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/kvpress/presses/snapkv_press.py b/kvpress/presses/snapkv_press.py index 371e9368..917dbb12 100644 --- a/kvpress/presses/snapkv_press.py +++ b/kvpress/presses/snapkv_press.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/kvpress/presses/streaming_llm_press.py b/kvpress/presses/streaming_llm_press.py index c63a85b8..5dae8de5 100644 --- a/kvpress/presses/streaming_llm_press.py +++ b/kvpress/presses/streaming_llm_press.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/kvpress/presses/think_press.py b/kvpress/presses/think_press.py index 55bf1d44..8e9cbf57 100644 --- a/kvpress/presses/think_press.py +++ b/kvpress/presses/think_press.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/kvpress/presses/tova_press.py b/kvpress/presses/tova_press.py index 5ca96cd1..e0d69180 100644 --- a/kvpress/presses/tova_press.py +++ b/kvpress/presses/tova_press.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/pyproject.toml b/pyproject.toml index 60dca287..56ee5532 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -2,7 +2,7 @@ name = "kvpress" authors = ["Simon Jegou", "Maximilian Jeblick", "Jiwei Liu", "David Austin"] description = "Efficiently compress the KV cache of any pretrained transformer" -version = "0.2.3" +version = "0.2.4" readme = "README.md" [tool.poetry.dependencies] diff --git a/tests/default_presses.py b/tests/default_presses.py index 4d699493..6ce5e6ef 100644 --- a/tests/default_presses.py +++ b/tests/default_presses.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 import numpy as np diff --git a/tests/fixtures.py b/tests/fixtures.py index 912c90fe..a08111a8 100644 --- a/tests/fixtures.py +++ b/tests/fixtures.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/tests/presses/test_key_rerotation_press_rope.py b/tests/presses/test_key_rerotation_press_rope.py index f890dc6f..a935b079 100644 --- a/tests/presses/test_key_rerotation_press_rope.py +++ b/tests/presses/test_key_rerotation_press_rope.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/tests/presses/test_observed_attention_press.py b/tests/presses/test_observed_attention_press.py index 7310a97c..287791c2 100644 --- a/tests/presses/test_observed_attention_press.py +++ b/tests/presses/test_observed_attention_press.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/tests/presses/test_presses.py b/tests/presses/test_presses.py index dfea2861..2adc9285 100644 --- a/tests/presses/test_presses.py +++ b/tests/presses/test_presses.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 from dataclasses import dataclass diff --git a/tests/test_generate.py b/tests/test_generate.py index c5315312..6a0b7066 100644 --- a/tests/test_generate.py +++ b/tests/test_generate.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/tests/test_per_layer_compression_press.py b/tests/test_per_layer_compression_press.py index 6db01eb2..78de3950 100644 --- a/tests/test_per_layer_compression_press.py +++ b/tests/test_per_layer_compression_press.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/tests/test_pipeline.py b/tests/test_pipeline.py index 5e371db5..14ff3156 100644 --- a/tests/test_pipeline.py +++ b/tests/test_pipeline.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 diff --git a/tests/test_press_call.py b/tests/test_press_call.py index 67a03328..d6307a8b 100644 --- a/tests/test_press_call.py +++ b/tests/test_press_call.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0