├── .gitignore ├── LICENSE ├── README.md ├── data ├── CoProv2_test.csv └── CoProv2_train.csv ├── environment.yaml ├── inference.py ├── launchers ├── train_lora_sd15.sh └── train_lora_sdxl.sh ├── test.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Visualignment/SafetyDPO/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Visualignment/SafetyDPO/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Visualignment/SafetyDPO/HEAD/README.md -------------------------------------------------------------------------------- /data/CoProv2_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Visualignment/SafetyDPO/HEAD/data/CoProv2_test.csv -------------------------------------------------------------------------------- /data/CoProv2_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Visualignment/SafetyDPO/HEAD/data/CoProv2_train.csv -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Visualignment/SafetyDPO/HEAD/environment.yaml -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Visualignment/SafetyDPO/HEAD/inference.py -------------------------------------------------------------------------------- /launchers/train_lora_sd15.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Visualignment/SafetyDPO/HEAD/launchers/train_lora_sd15.sh -------------------------------------------------------------------------------- /launchers/train_lora_sdxl.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Visualignment/SafetyDPO/HEAD/launchers/train_lora_sdxl.sh -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Visualignment/SafetyDPO/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Visualignment/SafetyDPO/HEAD/train.py --------------------------------------------------------------------------------