Views
No views yet
1from diffusers import StableDiffusionPipeline
2import torch
3
4pipe = StableDiffusionPipeline.from_pretrained("ErasureResearch/uce_garbage_truck", torch_dtype=torch.float16).to("cuda")
5prompt = "a photo of a garbage_truck"
6image = pipe(prompt).images[0]
7image.save("erased_garbage_truck.png")1@article{concept_erasure_2024,
2 title={Concept Erasure in Diffusion Models},
3 author={ErasureResearch Team},
4 journal={Proceedings of...},
5 year={2024}
6}