├── .gitignore ├── LICENSE ├── README.md ├── ag_news ├── build_dataset.py └── pred_eval.py ├── alpaca └── pred_eval.py ├── cache └── ignoreme ├── ckpt ├── ag_news │ └── ignoreme ├── gsm8k │ └── ignoreme ├── ignoreme └── sst2 │ └── ignoreme ├── gsm8k ├── build_dataset.py └── pred_eval.py ├── huggingface_token.txt ├── images └── vaccine_logo.png ├── poison └── evaluation │ ├── constants.py │ ├── eval_sentiment.py │ ├── moderation.py │ ├── pred.py │ └── utils.py ├── script ├── EWC_finetune │ ├── agnews.sh │ ├── gsm8k.sh │ └── sst2.sh ├── KL_finetune │ ├── agnews.sh │ ├── alpaca.sh │ ├── gsm8k.sh │ └── sst2.sh ├── alignment │ ├── SFT.sh │ ├── Vaccine.sh │ └── random_Vaccine.sh ├── no_alignment_finetune │ ├── agnews.sh │ ├── alpaca.sh │ ├── gsm8k.sh │ ├── single_lora_agnews.sh │ ├── single_lora_gsm8k.sh │ ├── single_lora_sst2.sh │ ├── sst2.sh │ ├── sst2_pretrain.sh │ ├── track_drift_sample_gsm8k.sh │ └── track_drift_sample_sst2.sh ├── random_vaccine_finetune │ ├── sst2.sh │ └── track_drift.sh ├── sft_finetune │ ├── agnews.sh │ ├── alpaca.sh │ ├── gsm8k.sh │ ├── single_lora_agnews.sh │ ├── single_lora_gsm8k.sh │ ├── single_lora_sst2.sh │ ├── sst2.sh │ ├── track_drift_sample_gsm8k.sh │ └── track_drift_sample_sst.sh ├── vaccine_finetune │ ├── Vaccine_EWC_agnews.sh │ ├── Vaccine_EWC_gsm8k.sh │ ├── Vaccine_EWC_sst2.sh │ ├── Vaccine_pre_train.sh │ ├── Vaccine_pre_train_agnews.sh │ ├── Vaccine_pre_train_gsm8k.sh │ ├── ag_news.sh │ ├── alpaca.sh │ ├── gsm8k.sh │ ├── rho_experiment.sh │ ├── single_lora_agnews.sh │ ├── single_lora_gsm8k.sh │ ├── single_lora_sst2.sh │ ├── sst2.sh │ └── track_drift.sh └── vlguard_finetune │ └── vlguard_poison_ratio.sh ├── sst2 ├── build_dataset.py └── pred_eval.py ├── train.py ├── trainer.py ├── utils.py ├── vaccine.yml └── vaccine_pip.txt /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/README.md -------------------------------------------------------------------------------- /ag_news/build_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/ag_news/build_dataset.py -------------------------------------------------------------------------------- /ag_news/pred_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/ag_news/pred_eval.py -------------------------------------------------------------------------------- /alpaca/pred_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/alpaca/pred_eval.py -------------------------------------------------------------------------------- /cache/ignoreme: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ckpt/ag_news/ignoreme: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ckpt/gsm8k/ignoreme: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ckpt/ignoreme: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ckpt/sst2/ignoreme: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gsm8k/build_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/gsm8k/build_dataset.py -------------------------------------------------------------------------------- /gsm8k/pred_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/gsm8k/pred_eval.py -------------------------------------------------------------------------------- /huggingface_token.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/huggingface_token.txt -------------------------------------------------------------------------------- /images/vaccine_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/images/vaccine_logo.png -------------------------------------------------------------------------------- /poison/evaluation/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/poison/evaluation/constants.py -------------------------------------------------------------------------------- /poison/evaluation/eval_sentiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/poison/evaluation/eval_sentiment.py -------------------------------------------------------------------------------- /poison/evaluation/moderation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/poison/evaluation/moderation.py -------------------------------------------------------------------------------- /poison/evaluation/pred.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/poison/evaluation/pred.py -------------------------------------------------------------------------------- /poison/evaluation/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/poison/evaluation/utils.py -------------------------------------------------------------------------------- /script/EWC_finetune/agnews.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/EWC_finetune/agnews.sh -------------------------------------------------------------------------------- /script/EWC_finetune/gsm8k.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/EWC_finetune/gsm8k.sh -------------------------------------------------------------------------------- /script/EWC_finetune/sst2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/EWC_finetune/sst2.sh -------------------------------------------------------------------------------- /script/KL_finetune/agnews.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/KL_finetune/agnews.sh -------------------------------------------------------------------------------- /script/KL_finetune/alpaca.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/KL_finetune/alpaca.sh -------------------------------------------------------------------------------- /script/KL_finetune/gsm8k.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/KL_finetune/gsm8k.sh -------------------------------------------------------------------------------- /script/KL_finetune/sst2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/KL_finetune/sst2.sh -------------------------------------------------------------------------------- /script/alignment/SFT.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/alignment/SFT.sh -------------------------------------------------------------------------------- /script/alignment/Vaccine.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/alignment/Vaccine.sh -------------------------------------------------------------------------------- /script/alignment/random_Vaccine.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/alignment/random_Vaccine.sh -------------------------------------------------------------------------------- /script/no_alignment_finetune/agnews.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/no_alignment_finetune/agnews.sh -------------------------------------------------------------------------------- /script/no_alignment_finetune/alpaca.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/no_alignment_finetune/alpaca.sh -------------------------------------------------------------------------------- /script/no_alignment_finetune/gsm8k.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/no_alignment_finetune/gsm8k.sh -------------------------------------------------------------------------------- /script/no_alignment_finetune/single_lora_agnews.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/no_alignment_finetune/single_lora_agnews.sh -------------------------------------------------------------------------------- /script/no_alignment_finetune/single_lora_gsm8k.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/no_alignment_finetune/single_lora_gsm8k.sh -------------------------------------------------------------------------------- /script/no_alignment_finetune/single_lora_sst2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/no_alignment_finetune/single_lora_sst2.sh -------------------------------------------------------------------------------- /script/no_alignment_finetune/sst2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/no_alignment_finetune/sst2.sh -------------------------------------------------------------------------------- /script/no_alignment_finetune/sst2_pretrain.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/no_alignment_finetune/sst2_pretrain.sh -------------------------------------------------------------------------------- /script/no_alignment_finetune/track_drift_sample_gsm8k.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/no_alignment_finetune/track_drift_sample_gsm8k.sh -------------------------------------------------------------------------------- /script/no_alignment_finetune/track_drift_sample_sst2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/no_alignment_finetune/track_drift_sample_sst2.sh -------------------------------------------------------------------------------- /script/random_vaccine_finetune/sst2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/random_vaccine_finetune/sst2.sh -------------------------------------------------------------------------------- /script/random_vaccine_finetune/track_drift.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/random_vaccine_finetune/track_drift.sh -------------------------------------------------------------------------------- /script/sft_finetune/agnews.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/sft_finetune/agnews.sh -------------------------------------------------------------------------------- /script/sft_finetune/alpaca.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/sft_finetune/alpaca.sh -------------------------------------------------------------------------------- /script/sft_finetune/gsm8k.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/sft_finetune/gsm8k.sh -------------------------------------------------------------------------------- /script/sft_finetune/single_lora_agnews.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/sft_finetune/single_lora_agnews.sh -------------------------------------------------------------------------------- /script/sft_finetune/single_lora_gsm8k.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/sft_finetune/single_lora_gsm8k.sh -------------------------------------------------------------------------------- /script/sft_finetune/single_lora_sst2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/sft_finetune/single_lora_sst2.sh -------------------------------------------------------------------------------- /script/sft_finetune/sst2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/sft_finetune/sst2.sh -------------------------------------------------------------------------------- /script/sft_finetune/track_drift_sample_gsm8k.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/sft_finetune/track_drift_sample_gsm8k.sh -------------------------------------------------------------------------------- /script/sft_finetune/track_drift_sample_sst.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/sft_finetune/track_drift_sample_sst.sh -------------------------------------------------------------------------------- /script/vaccine_finetune/Vaccine_EWC_agnews.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/vaccine_finetune/Vaccine_EWC_agnews.sh -------------------------------------------------------------------------------- /script/vaccine_finetune/Vaccine_EWC_gsm8k.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/vaccine_finetune/Vaccine_EWC_gsm8k.sh -------------------------------------------------------------------------------- /script/vaccine_finetune/Vaccine_EWC_sst2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/vaccine_finetune/Vaccine_EWC_sst2.sh -------------------------------------------------------------------------------- /script/vaccine_finetune/Vaccine_pre_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/vaccine_finetune/Vaccine_pre_train.sh -------------------------------------------------------------------------------- /script/vaccine_finetune/Vaccine_pre_train_agnews.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/vaccine_finetune/Vaccine_pre_train_agnews.sh -------------------------------------------------------------------------------- /script/vaccine_finetune/Vaccine_pre_train_gsm8k.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/vaccine_finetune/Vaccine_pre_train_gsm8k.sh -------------------------------------------------------------------------------- /script/vaccine_finetune/ag_news.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/vaccine_finetune/ag_news.sh -------------------------------------------------------------------------------- /script/vaccine_finetune/alpaca.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/vaccine_finetune/alpaca.sh -------------------------------------------------------------------------------- /script/vaccine_finetune/gsm8k.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/vaccine_finetune/gsm8k.sh -------------------------------------------------------------------------------- /script/vaccine_finetune/rho_experiment.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/vaccine_finetune/rho_experiment.sh -------------------------------------------------------------------------------- /script/vaccine_finetune/single_lora_agnews.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/vaccine_finetune/single_lora_agnews.sh -------------------------------------------------------------------------------- /script/vaccine_finetune/single_lora_gsm8k.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/vaccine_finetune/single_lora_gsm8k.sh -------------------------------------------------------------------------------- /script/vaccine_finetune/single_lora_sst2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/vaccine_finetune/single_lora_sst2.sh -------------------------------------------------------------------------------- /script/vaccine_finetune/sst2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/vaccine_finetune/sst2.sh -------------------------------------------------------------------------------- /script/vaccine_finetune/track_drift.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/vaccine_finetune/track_drift.sh -------------------------------------------------------------------------------- /script/vlguard_finetune/vlguard_poison_ratio.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/script/vlguard_finetune/vlguard_poison_ratio.sh -------------------------------------------------------------------------------- /sst2/build_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/sst2/build_dataset.py -------------------------------------------------------------------------------- /sst2/pred_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/sst2/pred_eval.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/train.py -------------------------------------------------------------------------------- /trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/trainer.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/utils.py -------------------------------------------------------------------------------- /vaccine.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/vaccine.yml -------------------------------------------------------------------------------- /vaccine_pip.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Vaccine/HEAD/vaccine_pip.txt --------------------------------------------------------------------------------