/activation_refilling

Primary LanguagePythonApache License 2.0Apache-2.0

Boosting Long-Context Information Seeking via Query-Guided Activation Refilling

For training

bash scripts/train_stage_1.sh
bash scripts/train_stage_2.sh

For evaluation

bash scripts/eval.sh