torch torchvision safetensors diffusers==0.21.1 transformers lycoris_lora flatten_json pyyaml oyaml tensorboard kornia invisible-watermark einops accelerate toml albumentations pydantic omegaconf k-diffusion open_clip_torch timm prodigyopt controlnet_aux==0.0.7