[ControlNet Training] Remove safety from controlnet (#4180)

Remove safety from controlnet

[ControlNet Training] Remove safety from controlnet (#4180)
Remove safety from controlnet
d620070b · Patrick von Platen · GitHub · b7a6e34c · d620070b
Unverified Commit d620070b authored Jul 21, 2023 by Patrick von Platen Committed by GitHub Jul 21, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 1 deletion

examples/controlnet/train_controlnet_sdxl.py examples/controlnet/train_controlnet_sdxl.py +0 -1

No files found.
--- a/examples/controlnet/train_controlnet_sdxl.py
+++ b/examples/controlnet/train_controlnet_sdxl.py
@@ -84,7 +84,6 @@ def log_validation(vae, unet, controlnet, args, accelerator, weight_dtype, step)
        vae=vae,
        unet=unet,
        controlnet=controlnet,
-        safety_checker=None,
        revision=args.revision,
        torch_dtype=weight_dtype,
    )