Supported extensions: .ckpt or .safetensors