_grads_to_none \
 --mixed_precision fp16

Inference
	
The trained model can be run with the StableDiffusionControlNetPipeline.
Set base_model_path and controlnet_path to the values --pretrained_model_name_or_path and
--output_dir were respectively set to in the training script.


	Copied
from diffusers import StableDiffusionControlNetPipeline, ControlNetModel, UniPCMultistepScheduler
from diffusers.utils import load_image
import torch

base_mod