├── .gitignore ├── README.md ├── data ├── example │ └── usps-loss-above=0.85.json ├── prompts-negative-filtered-v2.txt ├── prompts-negative-filtered.txt ├── prompts-negative-raw-v2.txt └── prompts-negative-raw.txt ├── files ├── install_env.sh ├── openai_key.txt └── pipeline.png ├── scripts ├── build-search-space.sh ├── run-attack.sh └── run-improve.sh └── src ├── build_search_space.py ├── configs ├── algo │ ├── epo.yaml │ ├── gpo.yaml │ └── hybrid.yaml └── task │ ├── attack-synonym.yaml │ ├── improve-antonym.yaml │ └── improve-nplib.yaml ├── model ├── cliptextmodel.py ├── sd.py └── sd_pipeline.py ├── optimizers ├── common │ ├── prompt_optimizer.py │ ├── rmsprop.py │ └── search_space.py ├── evolution │ └── epo.py └── gradient │ ├── __init__.py │ └── gpo.py ├── prompts ├── __init__.py ├── chatgpt.py ├── library.py ├── paths.py └── pregen.py ├── run_dpo_diff.py └── utils ├── __init__.py ├── args_utils.py ├── exp_utils.py ├── log_utils.py └── task_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/README.md -------------------------------------------------------------------------------- /data/example/usps-loss-above=0.85.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/data/example/usps-loss-above=0.85.json -------------------------------------------------------------------------------- /data/prompts-negative-filtered-v2.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/data/prompts-negative-filtered-v2.txt -------------------------------------------------------------------------------- /data/prompts-negative-filtered.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/data/prompts-negative-filtered.txt -------------------------------------------------------------------------------- /data/prompts-negative-raw-v2.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/data/prompts-negative-raw-v2.txt -------------------------------------------------------------------------------- /data/prompts-negative-raw.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/data/prompts-negative-raw.txt -------------------------------------------------------------------------------- /files/install_env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/files/install_env.sh -------------------------------------------------------------------------------- /files/openai_key.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /files/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/files/pipeline.png -------------------------------------------------------------------------------- /scripts/build-search-space.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/scripts/build-search-space.sh -------------------------------------------------------------------------------- /scripts/run-attack.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/scripts/run-attack.sh -------------------------------------------------------------------------------- /scripts/run-improve.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/scripts/run-improve.sh -------------------------------------------------------------------------------- /src/build_search_space.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/src/build_search_space.py -------------------------------------------------------------------------------- /src/configs/algo/epo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/src/configs/algo/epo.yaml -------------------------------------------------------------------------------- /src/configs/algo/gpo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/src/configs/algo/gpo.yaml -------------------------------------------------------------------------------- /src/configs/algo/hybrid.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/src/configs/algo/hybrid.yaml -------------------------------------------------------------------------------- /src/configs/task/attack-synonym.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/src/configs/task/attack-synonym.yaml -------------------------------------------------------------------------------- /src/configs/task/improve-antonym.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/src/configs/task/improve-antonym.yaml -------------------------------------------------------------------------------- /src/configs/task/improve-nplib.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/src/configs/task/improve-nplib.yaml -------------------------------------------------------------------------------- /src/model/cliptextmodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/src/model/cliptextmodel.py -------------------------------------------------------------------------------- /src/model/sd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/src/model/sd.py -------------------------------------------------------------------------------- /src/model/sd_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/src/model/sd_pipeline.py -------------------------------------------------------------------------------- /src/optimizers/common/prompt_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/src/optimizers/common/prompt_optimizer.py -------------------------------------------------------------------------------- /src/optimizers/common/rmsprop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/src/optimizers/common/rmsprop.py -------------------------------------------------------------------------------- /src/optimizers/common/search_space.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/src/optimizers/common/search_space.py -------------------------------------------------------------------------------- /src/optimizers/evolution/epo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/src/optimizers/evolution/epo.py -------------------------------------------------------------------------------- /src/optimizers/gradient/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/optimizers/gradient/gpo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/src/optimizers/gradient/gpo.py -------------------------------------------------------------------------------- /src/prompts/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/src/prompts/__init__.py -------------------------------------------------------------------------------- /src/prompts/chatgpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/src/prompts/chatgpt.py -------------------------------------------------------------------------------- /src/prompts/library.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/src/prompts/library.py -------------------------------------------------------------------------------- /src/prompts/paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/src/prompts/paths.py -------------------------------------------------------------------------------- /src/prompts/pregen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/src/prompts/pregen.py -------------------------------------------------------------------------------- /src/run_dpo_diff.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/src/run_dpo_diff.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/src/utils/__init__.py -------------------------------------------------------------------------------- /src/utils/args_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/src/utils/args_utils.py -------------------------------------------------------------------------------- /src/utils/exp_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/src/utils/exp_utils.py -------------------------------------------------------------------------------- /src/utils/log_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/src/utils/log_utils.py -------------------------------------------------------------------------------- /src/utils/task_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruocwang/dpo-diffusion/HEAD/src/utils/task_utils.py --------------------------------------------------------------------------------