├── AlphaEdit ├── AlphaEdit_hparams.py ├── AlphaEdit_main.py ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-39.pyc │ ├── compute_ks.cpython-39.pyc │ ├── compute_z.cpython-39.pyc │ ├── compute_z_v1.cpython-39.pyc │ ├── memit_cc_main.cpython-39.pyc │ ├── memit_hparams.cpython-39.pyc │ ├── memit_main.cpython-39.pyc │ ├── memit_main_test.cpython-39.pyc │ ├── memit_main_v1.cpython-39.pyc │ ├── memit_main_v2.cpython-39.pyc │ ├── memit_main_v3.cpython-39.pyc │ ├── ncmit_hparams.cpython-39.pyc │ └── ncmit_main.cpython-39.pyc ├── compute_ks.py └── compute_z.py ├── LICENSE ├── README.md ├── baselines ├── README.md ├── ft │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-39.pyc │ │ ├── ft_hparams.cpython-39.pyc │ │ └── ft_main.cpython-39.pyc │ ├── ft_hparams.py │ └── ft_main.py └── mend │ ├── README.md │ ├── __init__.py │ ├── __pycache__ │ ├── __init__.cpython-39.pyc │ ├── editable_model.cpython-39.pyc │ ├── hooks.cpython-39.pyc │ ├── losses.cpython-39.pyc │ ├── mend_hparams.cpython-39.pyc │ ├── mend_main.cpython-39.pyc │ ├── nn.cpython-39.pyc │ └── utils.cpython-39.pyc │ ├── algs │ ├── __pycache__ │ │ └── mend.cpython-39.pyc │ ├── enn.py │ ├── ft.py │ └── mend.py │ ├── config │ ├── alg │ │ ├── efk.yaml │ │ ├── enn.yaml │ │ ├── ft.yaml │ │ └── mend.yaml │ ├── config.yaml │ ├── experiment │ │ ├── fc.yaml │ │ ├── gen.yaml │ │ └── qa.yaml │ └── model │ │ ├── bart-base.yaml │ │ ├── bert-base.yaml │ │ ├── distilgpt2.yaml │ │ ├── gpt2.yaml │ │ ├── gpt2large.yaml │ │ ├── gpt2medium.yaml │ │ ├── gpt2xl.yaml │ │ ├── gptj.yaml │ │ ├── gptneo27.yaml │ │ ├── t5large.yaml │ │ ├── t5small.yaml │ │ ├── t5xl.yaml │ │ └── t5xxl.yaml │ ├── data_classes │ ├── fever.py │ ├── nq.py │ ├── wiki.py │ └── zsre.py │ ├── editable_model.py │ ├── hooks.py │ ├── losses.py │ ├── mend_hparams.py │ ├── mend_main.py │ ├── models.py │ ├── nn.py │ ├── oracle.py │ ├── requirements.txt │ ├── run.py │ ├── trainer.py │ └── utils.py ├── dsets ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-39.pyc │ ├── attr_snippets.cpython-39.pyc │ ├── counterfact.cpython-39.pyc │ ├── knowns.cpython-39.pyc │ ├── tfidf_stats.cpython-39.pyc │ └── zsre.cpython-39.pyc ├── attr_snippets.py ├── counterfact.py ├── knowns.py ├── mquake.py ├── tfidf_stats.py └── zsre.py ├── experiments ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-38.pyc │ ├── __init__.cpython-39.pyc │ ├── evaluate.cpython-38.pyc │ ├── evaluate.cpython-39.pyc │ ├── evaluate_sequential.cpython-38.pyc │ ├── evaluate_sequential.cpython-39.pyc │ ├── evaluate_unified.cpython-39.pyc │ ├── summarize.cpython-39.pyc │ └── summarize_sequential.cpython-39.pyc ├── causal_trace.py ├── evaluate.py ├── py │ ├── __pycache__ │ │ ├── eval_utils_counterfact.cpython-39.pyc │ │ └── eval_utils_zsre.cpython-39.pyc │ ├── demo.py │ ├── eval_utils_counterfact.py │ ├── eval_utils_mquake.py │ └── eval_utils_zsre.py ├── summarize.py └── sweep.py ├── globals.yml ├── glue_eval ├── __pycache__ │ ├── cola_eval.cpython-39.pyc │ ├── dialogue_eval.cpython-39.pyc │ ├── glue_eval.cpython-39.pyc │ ├── mmlu_eval.cpython-39.pyc │ ├── mrpc_eval.cpython-39.pyc │ ├── nli_eval.cpython-39.pyc │ ├── rte_eval.cpython-39.pyc │ ├── sentiment_analysis_eval.cpython-39.pyc │ ├── sst_eval.cpython-39.pyc │ └── useful_functions.cpython-39.pyc ├── cola_eval.py ├── dataset │ ├── __pycache__ │ │ └── useful_functions.cpython-39.pyc │ ├── cola.pkl │ ├── create_glue_subset.py │ ├── dialogue.pkl │ ├── mmlu.pkl │ ├── mrpc.pkl │ ├── nli.pkl │ ├── rte.pkl │ ├── sentiment_analysis.pkl │ ├── sst2.pkl │ └── useful_functions.py ├── dialogue_eval.py ├── glue_eval.py ├── mmlu_eval.py ├── mrpc_eval.py ├── nli_eval.py ├── rte_eval.py ├── sentiment_analysis_eval.py ├── sst_eval.py └── useful_functions.py ├── hparams ├── AlphaEdit │ ├── EleutherAI_gpt-j-6B.json │ ├── Llama3-8B.json │ ├── gpt2-xl.json │ └── phi-1.5.json ├── FT │ ├── EleutherAI_gpt-j-6B_constr.json │ ├── EleutherAI_gpt-j-6B_unconstr.json │ ├── EleutherAI_gpt-j-6B_wd.json │ ├── Llama3-8B.json │ ├── gpt2-large_constr.json │ ├── gpt2-medium_constr.json │ ├── gpt2-xl_attn.json │ ├── gpt2-xl_constr.json │ └── gpt2-xl_unconstr.json ├── MEMIT │ ├── EleutherAI_gpt-j-6B.json │ ├── Llama3-8B.json │ └── gpt2-xl.json ├── MEND │ ├── EleutherAI_gpt-j-6B.json │ ├── EleutherAI_gpt-j-6B_CF.json │ ├── gpt2-xl.json │ ├── gpt2-xl_CF.json │ └── gpt2-xl_zsRE.json ├── NSE │ ├── EleutherAI_gpt-j-6B.json │ ├── Llama3-8B.json │ └── gpt2-xl.json └── ROME │ ├── EleutherAI_gpt-j-6B.json │ ├── EleutherAI_gpt-neox-20b.json │ ├── Llama3-8B.json │ ├── gpt2-large.json │ ├── gpt2-medium.json │ └── gpt2-xl.json ├── memit ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-39.pyc │ ├── compute_ks.cpython-39.pyc │ ├── compute_z.cpython-39.pyc │ ├── compute_z_v1.cpython-39.pyc │ ├── memit_cc_main.cpython-39.pyc │ ├── memit_hparams.cpython-39.pyc │ ├── memit_main.cpython-39.pyc │ ├── memit_main_test.cpython-39.pyc │ ├── memit_main_v1.cpython-39.pyc │ ├── memit_main_v2.cpython-39.pyc │ ├── memit_main_v3.cpython-39.pyc │ ├── memit_rect_main.cpython-39.pyc │ └── memit_seq_main.cpython-39.pyc ├── compute_ks.py ├── compute_z.py ├── memit_hparams.py ├── memit_main.py ├── memit_rect_main.py └── memit_seq_main.py ├── nse ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-39.pyc │ ├── compute_ks.cpython-39.pyc │ ├── compute_z.cpython-39.pyc │ ├── memit_hparams.cpython-39.pyc │ ├── memit_main.cpython-39.pyc │ ├── memit_main_v1.cpython-39.pyc │ ├── nse_hparams.cpython-39.pyc │ └── nse_main.cpython-39.pyc ├── compute_ks.py ├── compute_z.py ├── nse_hparams.py └── nse_main.py ├── resource ├── alphaedit_fig.png ├── intro_fig.png └── model_fig.png ├── rome ├── README.md ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-39.pyc │ ├── compute_u.cpython-39.pyc │ ├── compute_v.cpython-39.pyc │ ├── layer_stats.cpython-38.pyc │ ├── layer_stats.cpython-39.pyc │ ├── repr_tools.cpython-39.pyc │ ├── rome_hparams.cpython-39.pyc │ ├── rome_main.cpython-39.pyc │ └── tok_dataset.cpython-39.pyc ├── compute_u.py ├── compute_v.py ├── layer_stats.py ├── repr_tools.py ├── rome_hparams.py ├── rome_main.py └── tok_dataset.py └── util ├── __init__.py ├── __pycache__ ├── __init__.cpython-39.pyc ├── generate.cpython-39.pyc ├── globals.cpython-39.pyc ├── hparams.cpython-39.pyc ├── logit_lens.cpython-39.pyc ├── nethook.cpython-39.pyc ├── perplexity.cpython-39.pyc └── runningstats.cpython-39.pyc ├── generate.py ├── globals.py ├── hparams.py ├── logit_lens.py ├── nethook.py ├── perplexity.py └── runningstats.py /AlphaEdit/AlphaEdit_hparams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/AlphaEdit/AlphaEdit_hparams.py -------------------------------------------------------------------------------- /AlphaEdit/AlphaEdit_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/AlphaEdit/AlphaEdit_main.py -------------------------------------------------------------------------------- /AlphaEdit/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/AlphaEdit/__init__.py -------------------------------------------------------------------------------- /AlphaEdit/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/AlphaEdit/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /AlphaEdit/__pycache__/compute_ks.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/AlphaEdit/__pycache__/compute_ks.cpython-39.pyc -------------------------------------------------------------------------------- /AlphaEdit/__pycache__/compute_z.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/AlphaEdit/__pycache__/compute_z.cpython-39.pyc -------------------------------------------------------------------------------- /AlphaEdit/__pycache__/compute_z_v1.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/AlphaEdit/__pycache__/compute_z_v1.cpython-39.pyc -------------------------------------------------------------------------------- /AlphaEdit/__pycache__/memit_cc_main.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/AlphaEdit/__pycache__/memit_cc_main.cpython-39.pyc -------------------------------------------------------------------------------- /AlphaEdit/__pycache__/memit_hparams.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/AlphaEdit/__pycache__/memit_hparams.cpython-39.pyc -------------------------------------------------------------------------------- /AlphaEdit/__pycache__/memit_main.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/AlphaEdit/__pycache__/memit_main.cpython-39.pyc -------------------------------------------------------------------------------- /AlphaEdit/__pycache__/memit_main_test.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/AlphaEdit/__pycache__/memit_main_test.cpython-39.pyc -------------------------------------------------------------------------------- /AlphaEdit/__pycache__/memit_main_v1.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/AlphaEdit/__pycache__/memit_main_v1.cpython-39.pyc -------------------------------------------------------------------------------- /AlphaEdit/__pycache__/memit_main_v2.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/AlphaEdit/__pycache__/memit_main_v2.cpython-39.pyc -------------------------------------------------------------------------------- /AlphaEdit/__pycache__/memit_main_v3.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/AlphaEdit/__pycache__/memit_main_v3.cpython-39.pyc -------------------------------------------------------------------------------- /AlphaEdit/__pycache__/ncmit_hparams.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/AlphaEdit/__pycache__/ncmit_hparams.cpython-39.pyc -------------------------------------------------------------------------------- /AlphaEdit/__pycache__/ncmit_main.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/AlphaEdit/__pycache__/ncmit_main.cpython-39.pyc -------------------------------------------------------------------------------- /AlphaEdit/compute_ks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/AlphaEdit/compute_ks.py -------------------------------------------------------------------------------- /AlphaEdit/compute_z.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/AlphaEdit/compute_z.py -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/README.md -------------------------------------------------------------------------------- /baselines/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/README.md -------------------------------------------------------------------------------- /baselines/ft/__init__.py: -------------------------------------------------------------------------------- 1 | from .ft_main import FTHyperParams, apply_ft_to_model, execute_ft 2 | -------------------------------------------------------------------------------- /baselines/ft/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/ft/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /baselines/ft/__pycache__/ft_hparams.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/ft/__pycache__/ft_hparams.cpython-39.pyc -------------------------------------------------------------------------------- /baselines/ft/__pycache__/ft_main.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/ft/__pycache__/ft_main.cpython-39.pyc -------------------------------------------------------------------------------- /baselines/ft/ft_hparams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/ft/ft_hparams.py -------------------------------------------------------------------------------- /baselines/ft/ft_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/ft/ft_main.py -------------------------------------------------------------------------------- /baselines/mend/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/README.md -------------------------------------------------------------------------------- /baselines/mend/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/__init__.py -------------------------------------------------------------------------------- /baselines/mend/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /baselines/mend/__pycache__/editable_model.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/__pycache__/editable_model.cpython-39.pyc -------------------------------------------------------------------------------- /baselines/mend/__pycache__/hooks.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/__pycache__/hooks.cpython-39.pyc -------------------------------------------------------------------------------- /baselines/mend/__pycache__/losses.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/__pycache__/losses.cpython-39.pyc -------------------------------------------------------------------------------- /baselines/mend/__pycache__/mend_hparams.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/__pycache__/mend_hparams.cpython-39.pyc -------------------------------------------------------------------------------- /baselines/mend/__pycache__/mend_main.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/__pycache__/mend_main.cpython-39.pyc -------------------------------------------------------------------------------- /baselines/mend/__pycache__/nn.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/__pycache__/nn.cpython-39.pyc -------------------------------------------------------------------------------- /baselines/mend/__pycache__/utils.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/__pycache__/utils.cpython-39.pyc -------------------------------------------------------------------------------- /baselines/mend/algs/__pycache__/mend.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/algs/__pycache__/mend.cpython-39.pyc -------------------------------------------------------------------------------- /baselines/mend/algs/enn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/algs/enn.py -------------------------------------------------------------------------------- /baselines/mend/algs/ft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/algs/ft.py -------------------------------------------------------------------------------- /baselines/mend/algs/mend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/algs/mend.py -------------------------------------------------------------------------------- /baselines/mend/config/alg/efk.yaml: -------------------------------------------------------------------------------- 1 | # @package _global_ 2 | 3 | alg: efk 4 | train_base: False 5 | lr: 1e-5 6 | -------------------------------------------------------------------------------- /baselines/mend/config/alg/enn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/config/alg/enn.yaml -------------------------------------------------------------------------------- /baselines/mend/config/alg/ft.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/config/alg/ft.yaml -------------------------------------------------------------------------------- /baselines/mend/config/alg/mend.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/config/alg/mend.yaml -------------------------------------------------------------------------------- /baselines/mend/config/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/config/config.yaml -------------------------------------------------------------------------------- /baselines/mend/config/experiment/fc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/config/experiment/fc.yaml -------------------------------------------------------------------------------- /baselines/mend/config/experiment/gen.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/config/experiment/gen.yaml -------------------------------------------------------------------------------- /baselines/mend/config/experiment/qa.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/config/experiment/qa.yaml -------------------------------------------------------------------------------- /baselines/mend/config/model/bart-base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/config/model/bart-base.yaml -------------------------------------------------------------------------------- /baselines/mend/config/model/bert-base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/config/model/bert-base.yaml -------------------------------------------------------------------------------- /baselines/mend/config/model/distilgpt2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/config/model/distilgpt2.yaml -------------------------------------------------------------------------------- /baselines/mend/config/model/gpt2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/config/model/gpt2.yaml -------------------------------------------------------------------------------- /baselines/mend/config/model/gpt2large.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/config/model/gpt2large.yaml -------------------------------------------------------------------------------- /baselines/mend/config/model/gpt2medium.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/config/model/gpt2medium.yaml -------------------------------------------------------------------------------- /baselines/mend/config/model/gpt2xl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/config/model/gpt2xl.yaml -------------------------------------------------------------------------------- /baselines/mend/config/model/gptj.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/config/model/gptj.yaml -------------------------------------------------------------------------------- /baselines/mend/config/model/gptneo27.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/config/model/gptneo27.yaml -------------------------------------------------------------------------------- /baselines/mend/config/model/t5large.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/config/model/t5large.yaml -------------------------------------------------------------------------------- /baselines/mend/config/model/t5small.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/config/model/t5small.yaml -------------------------------------------------------------------------------- /baselines/mend/config/model/t5xl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/config/model/t5xl.yaml -------------------------------------------------------------------------------- /baselines/mend/config/model/t5xxl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/config/model/t5xxl.yaml -------------------------------------------------------------------------------- /baselines/mend/data_classes/fever.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/data_classes/fever.py -------------------------------------------------------------------------------- /baselines/mend/data_classes/nq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/data_classes/nq.py -------------------------------------------------------------------------------- /baselines/mend/data_classes/wiki.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/data_classes/wiki.py -------------------------------------------------------------------------------- /baselines/mend/data_classes/zsre.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/data_classes/zsre.py -------------------------------------------------------------------------------- /baselines/mend/editable_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/editable_model.py -------------------------------------------------------------------------------- /baselines/mend/hooks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/hooks.py -------------------------------------------------------------------------------- /baselines/mend/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/losses.py -------------------------------------------------------------------------------- /baselines/mend/mend_hparams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/mend_hparams.py -------------------------------------------------------------------------------- /baselines/mend/mend_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/mend_main.py -------------------------------------------------------------------------------- /baselines/mend/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/models.py -------------------------------------------------------------------------------- /baselines/mend/nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/nn.py -------------------------------------------------------------------------------- /baselines/mend/oracle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/oracle.py -------------------------------------------------------------------------------- /baselines/mend/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/requirements.txt -------------------------------------------------------------------------------- /baselines/mend/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/run.py -------------------------------------------------------------------------------- /baselines/mend/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/trainer.py -------------------------------------------------------------------------------- /baselines/mend/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/baselines/mend/utils.py -------------------------------------------------------------------------------- /dsets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/dsets/__init__.py -------------------------------------------------------------------------------- /dsets/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/dsets/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /dsets/__pycache__/attr_snippets.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/dsets/__pycache__/attr_snippets.cpython-39.pyc -------------------------------------------------------------------------------- /dsets/__pycache__/counterfact.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/dsets/__pycache__/counterfact.cpython-39.pyc -------------------------------------------------------------------------------- /dsets/__pycache__/knowns.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/dsets/__pycache__/knowns.cpython-39.pyc -------------------------------------------------------------------------------- /dsets/__pycache__/tfidf_stats.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/dsets/__pycache__/tfidf_stats.cpython-39.pyc -------------------------------------------------------------------------------- /dsets/__pycache__/zsre.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/dsets/__pycache__/zsre.cpython-39.pyc -------------------------------------------------------------------------------- /dsets/attr_snippets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/dsets/attr_snippets.py -------------------------------------------------------------------------------- /dsets/counterfact.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/dsets/counterfact.py -------------------------------------------------------------------------------- /dsets/knowns.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/dsets/knowns.py -------------------------------------------------------------------------------- /dsets/mquake.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/dsets/mquake.py -------------------------------------------------------------------------------- /dsets/tfidf_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/dsets/tfidf_stats.py -------------------------------------------------------------------------------- /dsets/zsre.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/dsets/zsre.py -------------------------------------------------------------------------------- /experiments/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/experiments/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /experiments/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/experiments/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /experiments/__pycache__/evaluate.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/experiments/__pycache__/evaluate.cpython-38.pyc -------------------------------------------------------------------------------- /experiments/__pycache__/evaluate.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/experiments/__pycache__/evaluate.cpython-39.pyc -------------------------------------------------------------------------------- /experiments/__pycache__/evaluate_sequential.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/experiments/__pycache__/evaluate_sequential.cpython-38.pyc -------------------------------------------------------------------------------- /experiments/__pycache__/evaluate_sequential.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/experiments/__pycache__/evaluate_sequential.cpython-39.pyc -------------------------------------------------------------------------------- /experiments/__pycache__/evaluate_unified.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/experiments/__pycache__/evaluate_unified.cpython-39.pyc -------------------------------------------------------------------------------- /experiments/__pycache__/summarize.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/experiments/__pycache__/summarize.cpython-39.pyc -------------------------------------------------------------------------------- /experiments/__pycache__/summarize_sequential.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/experiments/__pycache__/summarize_sequential.cpython-39.pyc -------------------------------------------------------------------------------- /experiments/causal_trace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/experiments/causal_trace.py -------------------------------------------------------------------------------- /experiments/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/experiments/evaluate.py -------------------------------------------------------------------------------- /experiments/py/__pycache__/eval_utils_counterfact.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/experiments/py/__pycache__/eval_utils_counterfact.cpython-39.pyc -------------------------------------------------------------------------------- /experiments/py/__pycache__/eval_utils_zsre.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/experiments/py/__pycache__/eval_utils_zsre.cpython-39.pyc -------------------------------------------------------------------------------- /experiments/py/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/experiments/py/demo.py -------------------------------------------------------------------------------- /experiments/py/eval_utils_counterfact.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/experiments/py/eval_utils_counterfact.py -------------------------------------------------------------------------------- /experiments/py/eval_utils_mquake.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/experiments/py/eval_utils_mquake.py -------------------------------------------------------------------------------- /experiments/py/eval_utils_zsre.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/experiments/py/eval_utils_zsre.py -------------------------------------------------------------------------------- /experiments/summarize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/experiments/summarize.py -------------------------------------------------------------------------------- /experiments/sweep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/experiments/sweep.py -------------------------------------------------------------------------------- /globals.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/globals.yml -------------------------------------------------------------------------------- /glue_eval/__pycache__/cola_eval.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/__pycache__/cola_eval.cpython-39.pyc -------------------------------------------------------------------------------- /glue_eval/__pycache__/dialogue_eval.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/__pycache__/dialogue_eval.cpython-39.pyc -------------------------------------------------------------------------------- /glue_eval/__pycache__/glue_eval.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/__pycache__/glue_eval.cpython-39.pyc -------------------------------------------------------------------------------- /glue_eval/__pycache__/mmlu_eval.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/__pycache__/mmlu_eval.cpython-39.pyc -------------------------------------------------------------------------------- /glue_eval/__pycache__/mrpc_eval.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/__pycache__/mrpc_eval.cpython-39.pyc -------------------------------------------------------------------------------- /glue_eval/__pycache__/nli_eval.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/__pycache__/nli_eval.cpython-39.pyc -------------------------------------------------------------------------------- /glue_eval/__pycache__/rte_eval.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/__pycache__/rte_eval.cpython-39.pyc -------------------------------------------------------------------------------- /glue_eval/__pycache__/sentiment_analysis_eval.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/__pycache__/sentiment_analysis_eval.cpython-39.pyc -------------------------------------------------------------------------------- /glue_eval/__pycache__/sst_eval.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/__pycache__/sst_eval.cpython-39.pyc -------------------------------------------------------------------------------- /glue_eval/__pycache__/useful_functions.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/__pycache__/useful_functions.cpython-39.pyc -------------------------------------------------------------------------------- /glue_eval/cola_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/cola_eval.py -------------------------------------------------------------------------------- /glue_eval/dataset/__pycache__/useful_functions.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/dataset/__pycache__/useful_functions.cpython-39.pyc -------------------------------------------------------------------------------- /glue_eval/dataset/cola.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/dataset/cola.pkl -------------------------------------------------------------------------------- /glue_eval/dataset/create_glue_subset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/dataset/create_glue_subset.py -------------------------------------------------------------------------------- /glue_eval/dataset/dialogue.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/dataset/dialogue.pkl -------------------------------------------------------------------------------- /glue_eval/dataset/mmlu.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/dataset/mmlu.pkl -------------------------------------------------------------------------------- /glue_eval/dataset/mrpc.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/dataset/mrpc.pkl -------------------------------------------------------------------------------- /glue_eval/dataset/nli.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/dataset/nli.pkl -------------------------------------------------------------------------------- /glue_eval/dataset/rte.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/dataset/rte.pkl -------------------------------------------------------------------------------- /glue_eval/dataset/sentiment_analysis.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/dataset/sentiment_analysis.pkl -------------------------------------------------------------------------------- /glue_eval/dataset/sst2.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/dataset/sst2.pkl -------------------------------------------------------------------------------- /glue_eval/dataset/useful_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/dataset/useful_functions.py -------------------------------------------------------------------------------- /glue_eval/dialogue_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/dialogue_eval.py -------------------------------------------------------------------------------- /glue_eval/glue_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/glue_eval.py -------------------------------------------------------------------------------- /glue_eval/mmlu_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/mmlu_eval.py -------------------------------------------------------------------------------- /glue_eval/mrpc_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/mrpc_eval.py -------------------------------------------------------------------------------- /glue_eval/nli_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/nli_eval.py -------------------------------------------------------------------------------- /glue_eval/rte_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/rte_eval.py -------------------------------------------------------------------------------- /glue_eval/sentiment_analysis_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/sentiment_analysis_eval.py -------------------------------------------------------------------------------- /glue_eval/sst_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/sst_eval.py -------------------------------------------------------------------------------- /glue_eval/useful_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/glue_eval/useful_functions.py -------------------------------------------------------------------------------- /hparams/AlphaEdit/EleutherAI_gpt-j-6B.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/AlphaEdit/EleutherAI_gpt-j-6B.json -------------------------------------------------------------------------------- /hparams/AlphaEdit/Llama3-8B.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/AlphaEdit/Llama3-8B.json -------------------------------------------------------------------------------- /hparams/AlphaEdit/gpt2-xl.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/AlphaEdit/gpt2-xl.json -------------------------------------------------------------------------------- /hparams/AlphaEdit/phi-1.5.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/AlphaEdit/phi-1.5.json -------------------------------------------------------------------------------- /hparams/FT/EleutherAI_gpt-j-6B_constr.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/FT/EleutherAI_gpt-j-6B_constr.json -------------------------------------------------------------------------------- /hparams/FT/EleutherAI_gpt-j-6B_unconstr.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/FT/EleutherAI_gpt-j-6B_unconstr.json -------------------------------------------------------------------------------- /hparams/FT/EleutherAI_gpt-j-6B_wd.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/FT/EleutherAI_gpt-j-6B_wd.json -------------------------------------------------------------------------------- /hparams/FT/Llama3-8B.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/FT/Llama3-8B.json -------------------------------------------------------------------------------- /hparams/FT/gpt2-large_constr.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/FT/gpt2-large_constr.json -------------------------------------------------------------------------------- /hparams/FT/gpt2-medium_constr.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/FT/gpt2-medium_constr.json -------------------------------------------------------------------------------- /hparams/FT/gpt2-xl_attn.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/FT/gpt2-xl_attn.json -------------------------------------------------------------------------------- /hparams/FT/gpt2-xl_constr.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/FT/gpt2-xl_constr.json -------------------------------------------------------------------------------- /hparams/FT/gpt2-xl_unconstr.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/FT/gpt2-xl_unconstr.json -------------------------------------------------------------------------------- /hparams/MEMIT/EleutherAI_gpt-j-6B.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/MEMIT/EleutherAI_gpt-j-6B.json -------------------------------------------------------------------------------- /hparams/MEMIT/Llama3-8B.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/MEMIT/Llama3-8B.json -------------------------------------------------------------------------------- /hparams/MEMIT/gpt2-xl.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/MEMIT/gpt2-xl.json -------------------------------------------------------------------------------- /hparams/MEND/EleutherAI_gpt-j-6B.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/MEND/EleutherAI_gpt-j-6B.json -------------------------------------------------------------------------------- /hparams/MEND/EleutherAI_gpt-j-6B_CF.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/MEND/EleutherAI_gpt-j-6B_CF.json -------------------------------------------------------------------------------- /hparams/MEND/gpt2-xl.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/MEND/gpt2-xl.json -------------------------------------------------------------------------------- /hparams/MEND/gpt2-xl_CF.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/MEND/gpt2-xl_CF.json -------------------------------------------------------------------------------- /hparams/MEND/gpt2-xl_zsRE.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/MEND/gpt2-xl_zsRE.json -------------------------------------------------------------------------------- /hparams/NSE/EleutherAI_gpt-j-6B.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/NSE/EleutherAI_gpt-j-6B.json -------------------------------------------------------------------------------- /hparams/NSE/Llama3-8B.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/NSE/Llama3-8B.json -------------------------------------------------------------------------------- /hparams/NSE/gpt2-xl.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/NSE/gpt2-xl.json -------------------------------------------------------------------------------- /hparams/ROME/EleutherAI_gpt-j-6B.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/ROME/EleutherAI_gpt-j-6B.json -------------------------------------------------------------------------------- /hparams/ROME/EleutherAI_gpt-neox-20b.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/ROME/EleutherAI_gpt-neox-20b.json -------------------------------------------------------------------------------- /hparams/ROME/Llama3-8B.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/ROME/Llama3-8B.json -------------------------------------------------------------------------------- /hparams/ROME/gpt2-large.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/ROME/gpt2-large.json -------------------------------------------------------------------------------- /hparams/ROME/gpt2-medium.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/ROME/gpt2-medium.json -------------------------------------------------------------------------------- /hparams/ROME/gpt2-xl.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/hparams/ROME/gpt2-xl.json -------------------------------------------------------------------------------- /memit/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/memit/__init__.py -------------------------------------------------------------------------------- /memit/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/memit/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /memit/__pycache__/compute_ks.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/memit/__pycache__/compute_ks.cpython-39.pyc -------------------------------------------------------------------------------- /memit/__pycache__/compute_z.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/memit/__pycache__/compute_z.cpython-39.pyc -------------------------------------------------------------------------------- /memit/__pycache__/compute_z_v1.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/memit/__pycache__/compute_z_v1.cpython-39.pyc -------------------------------------------------------------------------------- /memit/__pycache__/memit_cc_main.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/memit/__pycache__/memit_cc_main.cpython-39.pyc -------------------------------------------------------------------------------- /memit/__pycache__/memit_hparams.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/memit/__pycache__/memit_hparams.cpython-39.pyc -------------------------------------------------------------------------------- /memit/__pycache__/memit_main.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/memit/__pycache__/memit_main.cpython-39.pyc -------------------------------------------------------------------------------- /memit/__pycache__/memit_main_test.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/memit/__pycache__/memit_main_test.cpython-39.pyc -------------------------------------------------------------------------------- /memit/__pycache__/memit_main_v1.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/memit/__pycache__/memit_main_v1.cpython-39.pyc -------------------------------------------------------------------------------- /memit/__pycache__/memit_main_v2.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/memit/__pycache__/memit_main_v2.cpython-39.pyc -------------------------------------------------------------------------------- /memit/__pycache__/memit_main_v3.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/memit/__pycache__/memit_main_v3.cpython-39.pyc -------------------------------------------------------------------------------- /memit/__pycache__/memit_rect_main.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/memit/__pycache__/memit_rect_main.cpython-39.pyc -------------------------------------------------------------------------------- /memit/__pycache__/memit_seq_main.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/memit/__pycache__/memit_seq_main.cpython-39.pyc -------------------------------------------------------------------------------- /memit/compute_ks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/memit/compute_ks.py -------------------------------------------------------------------------------- /memit/compute_z.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/memit/compute_z.py -------------------------------------------------------------------------------- /memit/memit_hparams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/memit/memit_hparams.py -------------------------------------------------------------------------------- /memit/memit_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/memit/memit_main.py -------------------------------------------------------------------------------- /memit/memit_rect_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/memit/memit_rect_main.py -------------------------------------------------------------------------------- /memit/memit_seq_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/memit/memit_seq_main.py -------------------------------------------------------------------------------- /nse/__init__.py: -------------------------------------------------------------------------------- 1 | from .nse_main import NSEHyperParams, apply_nse_to_model -------------------------------------------------------------------------------- /nse/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/nse/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /nse/__pycache__/compute_ks.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/nse/__pycache__/compute_ks.cpython-39.pyc -------------------------------------------------------------------------------- /nse/__pycache__/compute_z.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/nse/__pycache__/compute_z.cpython-39.pyc -------------------------------------------------------------------------------- /nse/__pycache__/memit_hparams.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/nse/__pycache__/memit_hparams.cpython-39.pyc -------------------------------------------------------------------------------- /nse/__pycache__/memit_main.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/nse/__pycache__/memit_main.cpython-39.pyc -------------------------------------------------------------------------------- /nse/__pycache__/memit_main_v1.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/nse/__pycache__/memit_main_v1.cpython-39.pyc -------------------------------------------------------------------------------- /nse/__pycache__/nse_hparams.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/nse/__pycache__/nse_hparams.cpython-39.pyc -------------------------------------------------------------------------------- /nse/__pycache__/nse_main.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/nse/__pycache__/nse_main.cpython-39.pyc -------------------------------------------------------------------------------- /nse/compute_ks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/nse/compute_ks.py -------------------------------------------------------------------------------- /nse/compute_z.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/nse/compute_z.py -------------------------------------------------------------------------------- /nse/nse_hparams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/nse/nse_hparams.py -------------------------------------------------------------------------------- /nse/nse_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/nse/nse_main.py -------------------------------------------------------------------------------- /resource/alphaedit_fig.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/resource/alphaedit_fig.png -------------------------------------------------------------------------------- /resource/intro_fig.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/resource/intro_fig.png -------------------------------------------------------------------------------- /resource/model_fig.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/resource/model_fig.png -------------------------------------------------------------------------------- /rome/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/rome/README.md -------------------------------------------------------------------------------- /rome/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/rome/__init__.py -------------------------------------------------------------------------------- /rome/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/rome/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /rome/__pycache__/compute_u.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/rome/__pycache__/compute_u.cpython-39.pyc -------------------------------------------------------------------------------- /rome/__pycache__/compute_v.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/rome/__pycache__/compute_v.cpython-39.pyc -------------------------------------------------------------------------------- /rome/__pycache__/layer_stats.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/rome/__pycache__/layer_stats.cpython-38.pyc -------------------------------------------------------------------------------- /rome/__pycache__/layer_stats.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/rome/__pycache__/layer_stats.cpython-39.pyc -------------------------------------------------------------------------------- /rome/__pycache__/repr_tools.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/rome/__pycache__/repr_tools.cpython-39.pyc -------------------------------------------------------------------------------- /rome/__pycache__/rome_hparams.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/rome/__pycache__/rome_hparams.cpython-39.pyc -------------------------------------------------------------------------------- /rome/__pycache__/rome_main.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/rome/__pycache__/rome_main.cpython-39.pyc -------------------------------------------------------------------------------- /rome/__pycache__/tok_dataset.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/rome/__pycache__/tok_dataset.cpython-39.pyc -------------------------------------------------------------------------------- /rome/compute_u.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/rome/compute_u.py -------------------------------------------------------------------------------- /rome/compute_v.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/rome/compute_v.py -------------------------------------------------------------------------------- /rome/layer_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/rome/layer_stats.py -------------------------------------------------------------------------------- /rome/repr_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/rome/repr_tools.py -------------------------------------------------------------------------------- /rome/rome_hparams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/rome/rome_hparams.py -------------------------------------------------------------------------------- /rome/rome_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/rome/rome_main.py -------------------------------------------------------------------------------- /rome/tok_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/rome/tok_dataset.py -------------------------------------------------------------------------------- /util/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/util/__init__.py -------------------------------------------------------------------------------- /util/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/util/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /util/__pycache__/generate.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/util/__pycache__/generate.cpython-39.pyc -------------------------------------------------------------------------------- /util/__pycache__/globals.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/util/__pycache__/globals.cpython-39.pyc -------------------------------------------------------------------------------- /util/__pycache__/hparams.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/util/__pycache__/hparams.cpython-39.pyc -------------------------------------------------------------------------------- /util/__pycache__/logit_lens.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/util/__pycache__/logit_lens.cpython-39.pyc -------------------------------------------------------------------------------- /util/__pycache__/nethook.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/util/__pycache__/nethook.cpython-39.pyc -------------------------------------------------------------------------------- /util/__pycache__/perplexity.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/util/__pycache__/perplexity.cpython-39.pyc -------------------------------------------------------------------------------- /util/__pycache__/runningstats.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/util/__pycache__/runningstats.cpython-39.pyc -------------------------------------------------------------------------------- /util/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/util/generate.py -------------------------------------------------------------------------------- /util/globals.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/util/globals.py -------------------------------------------------------------------------------- /util/hparams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/util/hparams.py -------------------------------------------------------------------------------- /util/logit_lens.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/util/logit_lens.py -------------------------------------------------------------------------------- /util/nethook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/util/nethook.py -------------------------------------------------------------------------------- /util/perplexity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/util/perplexity.py -------------------------------------------------------------------------------- /util/runningstats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianghoucheng/AlphaEdit/HEAD/util/runningstats.py --------------------------------------------------------------------------------