# Trellis2 MLX requirements (Apple Silicon)
# NO: flash_attn, spconv, torchsparse (CUDA-only)

# Core
torch>=2.2.0
torchvision>=0.17.0
transformers>=4.40.0,<5
safetensors
huggingface_hub
pillow
numpy

# Mesh processing
trimesh
xatlas
fast-simplification
pygltflib

# Image processing
opencv-python-headless

# Server
fastapi
uvicorn[standard]
pydantic>=2.0

# MLX (Apple Silicon native)
mlx>=0.31.0

# Metal GPU packages (Apple Silicon — install with: pip install --no-build-isolation <url>)
mtldiffrast @ https://github.com/pedronaugusto/mtldiffrast/archive/main.tar.gz
cumesh @ https://github.com/pedronaugusto/mtlmesh/archive/main.tar.gz
flex_gemm @ https://github.com/pedronaugusto/mtlgemm/archive/main.tar.gz

# Misc
tqdm
imageio
imageio-ffmpeg
scipy
einops