ng in a bucket" \
  --report_to="wandb" \
  --push_to_hub  Once training is finished, you can use your new Custom Diffusion model for inference. single concept multiple concepts    Copied import torch
from diffusers import DiffusionPipeline

pipeline = DiffusionPipeline.from_pretrained(
    "CompVis/stable-diffusion-v1-4", torch_dtype=torch.float16,
).to("cuda")
pipeline.unet.load_attn_procs("path-to-save-model", weight_name="pytorch_custom_dif