torch
safetensors
pytorch-lightning
clip