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

patrickvonplaten · web-flow · commit d620070bb361 · 2023-07-21T08:03:59.000+05:30
Remove safety from controlnet
diff --git 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,
     )

Original file line number	Diff line number	Diff line change
`@@ -84,7 +84,6 @@ def log_validation(vae, unet, controlnet, args, accelerator, weight_dtype, step)`
`84`	`84`	`vae=vae,`
`85`	`85`	`unet=unet,`
`86`	`86`	`controlnet=controlnet,`
`87`		`- safety_checker=None,`
`88`	`87`	`revision=args.revision,`
`89`	`88`	`torch_dtype=weight_dtype,`
`90`	`89`	`)`