Add text-generation pipeline tag, link to code and license (#1)

- Add text-generation pipeline tag, link to code and license (99cff5c69181be68896d35ddb1a4f8aae7fba44b)


Co-authored-by: Niels Rogge <[email protected]>

README.md

@@ -1,7 +1,10 @@
 ---
 library_name: transformers
 tags: []
+pipeline_tag: text-generation
+license: mit
 ---
+
 **Repository for:**
 
 **ThinkEdit-deepseek-qwen-1.5b**
@@ -11,6 +14,8 @@ tags: []
 **Authors**: Chung-En Sun, Ge Yan, Tsui-Wei Weng\
 **Paper**: [ThinkEdit: Interpretable Weight Editing to Mitigate Overly Short Thinking in Reasoning Models](https://arxiv.org/abs/2503.22048)
 
+Github: https://github.com/Trustworthy-ML-Lab/ThinkEdit
+
 ---
 
 ## Introduction
@@ -19,8 +24,8 @@ Reasoning-augmented models sometimes fail by generating **overly short**, abstra
 
 **ThinkEdit** is a lightweight weight-editing method that:
 
-- Identifies \~2% of "short reasoning" attention heads
-- Edits only \~0.1% of total parameters
+- Identifies ~2% of "short reasoning" attention heads
+- Edits only ~0.1% of total parameters
 - Removes the "short reasoning" direction from their output
 - Boosts performance, especially on cases with short reasoning traces