Skip to content

GitLab

Unverified Commit d620070b authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

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

Remove safety from controlnet
parent b7a6e34c
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
examples/controlnet/train_controlnet_sdxl.py
View file @ d620070b
...@@ -84,7 +84,6 @@ def log_validation(vae, unet, controlnet, args, accelerator, weight_dtype, step) ...@@ -84,7 +84,6 @@ def log_validation(vae, unet, controlnet, args, accelerator, weight_dtype, step)
vae=vae, vae=vae,
unet=unet, unet=unet,
controlnet=controlnet, controlnet=controlnet,
safety_checker=None,
revision=args.revision, revision=args.revision,
torch_dtype=weight_dtype, torch_dtype=weight_dtype,
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment