|
|
--- |
|
|
license: mit |
|
|
--- |
|
|
|
|
|
# NSFW.pt used in HUB (https://github.com/ml-postech/HUB) |
|
|
|
|
|
It is generated using repo |
|
|
``` |
|
|
git clone https://github.com/rohitgandikota/unified-concept-editing.git |
|
|
cd unified-concept-editing |
|
|
mkdir models |
|
|
python3 -m venv .venv |
|
|
source .venv/bin/activate |
|
|
pip install -r requirements.txt |
|
|
``` |
|
|
Then |
|
|
``` |
|
|
python trainscripts/uce_sd_erase.py --model_id 'CompVis/stable-diffusion-v1-4' --edit_concepts 'nudity; porn; sex' --concept_type object --save_dir uce --exp_name NSFW --device cuda |
|
|
python -c "import torch; from safetensors.torch import load_file; torch.save(load_file('uce/NSFW.safetensors'), 'uce/NSFW.pt')" |
|
|
``` |
|
|
|
|
|
Note: |
|
|
The weights here are a partial UNet state_dict (e.g. only a few attn2.to_k/to_v weights). So in loading it with PyTorch’s default strict=False is required. |
