├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE ├── NUMBERS.md ├── README.md ├── main.py ├── prompt2output └── config.json ├── requirements.txt ├── setup.cfg ├── setup.py └── vec2text ├── __init__.py ├── aliases.py ├── analyze_utils.py ├── api.py ├── collator.py ├── data_helpers.py ├── experiments.py ├── generation_experiments.py ├── metrics.py ├── models ├── __init__.py ├── config.py ├── corrector_encoder.py ├── corrector_encoder_from_logits.py ├── inversion.py ├── inversion_bow.py ├── inversion_decoder.py ├── inversion_from_jailbreak.py ├── inversion_from_logits.py ├── inversion_from_logits_emb.py ├── inversion_from_logits_masked.py ├── inversion_from_multiple_logits.py ├── inversion_from_output.py ├── inversion_na.py ├── model_utils.py └── my_t5_base.py ├── precompute_train_hypotheses.py ├── prompts.py ├── run.py ├── run_args.py ├── tokenize_data.py ├── trainers ├── __init__.py ├── base.py ├── corrector.py ├── inversion.py ├── inversion_bow.py ├── inversion_from_logits.py └── inversion_na.py ├── trainers_baseline ├── __init__.py ├── decode_inversion_trainer.py ├── fewshot_inversion_trainer.py └── jailbreak_prompt_trainer.py ├── unigram.pt └── utils ├── __init__.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/LICENSE -------------------------------------------------------------------------------- /NUMBERS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/NUMBERS.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/README.md -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/main.py -------------------------------------------------------------------------------- /prompt2output/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/prompt2output/config.json -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/setup.py -------------------------------------------------------------------------------- /vec2text/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/__init__.py -------------------------------------------------------------------------------- /vec2text/aliases.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/aliases.py -------------------------------------------------------------------------------- /vec2text/analyze_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/analyze_utils.py -------------------------------------------------------------------------------- /vec2text/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/api.py -------------------------------------------------------------------------------- /vec2text/collator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/collator.py -------------------------------------------------------------------------------- /vec2text/data_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/data_helpers.py -------------------------------------------------------------------------------- /vec2text/experiments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/experiments.py -------------------------------------------------------------------------------- /vec2text/generation_experiments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/generation_experiments.py -------------------------------------------------------------------------------- /vec2text/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/metrics.py -------------------------------------------------------------------------------- /vec2text/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/models/__init__.py -------------------------------------------------------------------------------- /vec2text/models/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/models/config.py -------------------------------------------------------------------------------- /vec2text/models/corrector_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/models/corrector_encoder.py -------------------------------------------------------------------------------- /vec2text/models/corrector_encoder_from_logits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/models/corrector_encoder_from_logits.py -------------------------------------------------------------------------------- /vec2text/models/inversion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/models/inversion.py -------------------------------------------------------------------------------- /vec2text/models/inversion_bow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/models/inversion_bow.py -------------------------------------------------------------------------------- /vec2text/models/inversion_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/models/inversion_decoder.py -------------------------------------------------------------------------------- /vec2text/models/inversion_from_jailbreak.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/models/inversion_from_jailbreak.py -------------------------------------------------------------------------------- /vec2text/models/inversion_from_logits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/models/inversion_from_logits.py -------------------------------------------------------------------------------- /vec2text/models/inversion_from_logits_emb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/models/inversion_from_logits_emb.py -------------------------------------------------------------------------------- /vec2text/models/inversion_from_logits_masked.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/models/inversion_from_logits_masked.py -------------------------------------------------------------------------------- /vec2text/models/inversion_from_multiple_logits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/models/inversion_from_multiple_logits.py -------------------------------------------------------------------------------- /vec2text/models/inversion_from_output.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/models/inversion_from_output.py -------------------------------------------------------------------------------- /vec2text/models/inversion_na.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/models/inversion_na.py -------------------------------------------------------------------------------- /vec2text/models/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/models/model_utils.py -------------------------------------------------------------------------------- /vec2text/models/my_t5_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/models/my_t5_base.py -------------------------------------------------------------------------------- /vec2text/precompute_train_hypotheses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/precompute_train_hypotheses.py -------------------------------------------------------------------------------- /vec2text/prompts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/prompts.py -------------------------------------------------------------------------------- /vec2text/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/run.py -------------------------------------------------------------------------------- /vec2text/run_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/run_args.py -------------------------------------------------------------------------------- /vec2text/tokenize_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/tokenize_data.py -------------------------------------------------------------------------------- /vec2text/trainers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/trainers/__init__.py -------------------------------------------------------------------------------- /vec2text/trainers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/trainers/base.py -------------------------------------------------------------------------------- /vec2text/trainers/corrector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/trainers/corrector.py -------------------------------------------------------------------------------- /vec2text/trainers/inversion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/trainers/inversion.py -------------------------------------------------------------------------------- /vec2text/trainers/inversion_bow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/trainers/inversion_bow.py -------------------------------------------------------------------------------- /vec2text/trainers/inversion_from_logits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/trainers/inversion_from_logits.py -------------------------------------------------------------------------------- /vec2text/trainers/inversion_na.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/trainers/inversion_na.py -------------------------------------------------------------------------------- /vec2text/trainers_baseline/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/trainers_baseline/__init__.py -------------------------------------------------------------------------------- /vec2text/trainers_baseline/decode_inversion_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/trainers_baseline/decode_inversion_trainer.py -------------------------------------------------------------------------------- /vec2text/trainers_baseline/fewshot_inversion_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/trainers_baseline/fewshot_inversion_trainer.py -------------------------------------------------------------------------------- /vec2text/trainers_baseline/jailbreak_prompt_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/trainers_baseline/jailbreak_prompt_trainer.py -------------------------------------------------------------------------------- /vec2text/unigram.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/unigram.pt -------------------------------------------------------------------------------- /vec2text/utils/__init__.py: -------------------------------------------------------------------------------- 1 | from .utils import * # noqa: F401,F403 2 | -------------------------------------------------------------------------------- /vec2text/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/collinzrj/output2prompt/HEAD/vec2text/utils/utils.py --------------------------------------------------------------------------------