ParticleViT-XL / requirements.txt
jaluus's picture
Add requirements.txt with torch>=2.4 floor (nn.RMSNorm)
7041394 verified
Raw
History Blame Contribute Delete
296 Bytes
# Runtime dependencies for ParticleViT (PyTorch-only, no einops / flash-attn).
torch>=2.4 # nn.RMSNorm was added in 2.4; F.scaled_dot_product_attention needs >=2.0
safetensors>=0.4.0
huggingface_hub>=0.20
# Validated with: torch 2.12.0+cu130, safetensors 0.8.0, huggingface_hub 1.21.0