41 lines
765 B
Plaintext
41 lines
765 B
Plaintext
# 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
|