Add pipeline tag and improve model card

README.md

@@ -1,8 +1,9 @@
 ---
-license: apache-2.0
 datasets:
 - nvidia/Nemotron-Pretraining-Dataset-sample
 library_name: transformers
+license: apache-2.0
+pipeline_tag: other
 tags:
 - nvidia
 - pytorch
@@ -13,7 +14,7 @@ track_downloads: true
 
 
 [![License](https://img.shields.io/badge/License-Apache%202.0-green.svg)](https://www.apache.org/licenses/LICENSE-2.0)
-[![GitHub](https://img.shields.io/badge/GitHub-kvpress-blue?logo=github)](https://github.com/NVIDIA/kvpress/kvzap)
+[![GitHub](https://img.shields.io/badge/GitHub-kvpress-blue?logo=github)](https://github.com/NVIDIA/kvpress/tree/main/kvzap)
 [![KVzap collection](https://img.shields.io/badge/🤗%20Hugging%20Face-Collection-orange)](https://huggingface.co/collections/nvidia/kvzap)
 [![arXiv](https://img.shields.io/badge/arXiv-2601.07891-b31b1b.svg)](https://arxiv.org/abs/2601.07891)
 
@@ -21,4 +22,4 @@ track_downloads: true
 [KVzap](https://arxiv.org/abs/2601.07891) is a KV cache pruning method aiming to accelerate LLM inference in both prefilling and decoding. It applies a lightweight model to the hidden states to predict importance scores for every KV pair and prunes the ones with a score below a given threshold.
 
 
- KVzap is trained as a fast approximation of [KVzip+](https://arxiv.org/abs/2505.23416), using 1.2M samples from [Nemotron-Pretraining-Dataset-sample](https://huggingface.co/datasets/nvidia/Nemotron-Pretraining-Dataset-sample). Training code is available in the kvpress repository ([source](https://github.com/NVIDIA/kvpress/blob/main/kvzap)).
+ KVzap is trained as a fast approximation of [KVzip+](https://arxiv.org/abs/2505.23416), using 1.2M samples from [Nemotron-Pretraining-Dataset-sample](https://huggingface.co/datasets/nvidia/Nemotron-Pretraining-Dataset-sample). Training code is available in the kvpress repository ([source](https://github.com/NVIDIA/kvpress/blob/main/kvzap)).