torch torchdiffeq torchsde omegaconf einops open-clip-torch transformers safetensors pytorch_lightning accelerate