Skip to content

OneAndZero24/diffusion_unlearning

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

321 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

LoRA-SCD: LoRA-based Self-Censoring for Diffusion Models

Jan Miksa @ GMUM, Jagiellonian University

Work In Progress...

📦 Environment Setup

Python 3.12

python3 -m venv venv
source venv/bin/activate
pip install -r requirements.txt

Download the original Stable Diffusion model:

mkdir models
wget models/sd-v1-4.ckpt "https://huggingface.co/CompVis/stable-diffusion-v-1-4-original/resolve/main/sd-v1-4.ckpt?download=true"

🚀 Training

python train.py configs/pipeline.json

🖼️ Image Generation

python sample.py configs/pipeline.json 

About

LoRA-based Self-Censoring for Diffusion Models

Topics

Resources

Stars

Watchers

Forks

Contributors

Languages