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