File size: 233 Bytes
4811c23 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | # A100 训练环境依赖 (Python 3.10/3.11, CUDA 12.x) torch>=2.1 torchvision>=0.16 diffusers>=0.27 accelerate>=0.28 omegaconf>=2.3 einops lpips pyiqa opencv-python scipy numpy pillow safetensors huggingface_hub pyyaml requests |