10708-unlearning

Course Project for 10708 (Probabilistic Graphical Models): Unlearning in Diffusion Models

Commands for running:

python finetune.py --erase_concept "parachute" --train_method "xattn" --iterations 100 --lr 1e-5

python generate.py --erase_concept "parachute" --train_method "xattn" --output_dir tmp --finetuner --model_dir "models_new" --epochs 20

Name		Name	Last commit message	Last commit date
Latest commit History 3 Commits
erasing @ 6cdcefb		erasing @ 6cdcefb
output		output
upload_images		upload_images
.gitmodules		.gitmodules
README.md		README.md
accuracy.py		accuracy.py
classify.py		classify.py
finetune.py		finetune.py
generate.py		generate.py