Update requirements.txt

requirements.txt

@@ -1,30 +1,143 @@
-# --- Core Dependencies ---
-torch
-torchvision
-accelerate==1.1.1
-diffusers==0.30.0
+# Lyra 2 Requirements
+# Python 3.10+
+#
+# Install with: pip install --no-deps -r requirements.txt
+# MoGe must be installed separately (needs its own deps):
+#   pip install "git+https://github.com/microsoft/MoGe.git"
 
-einops==0.8.1
-#flash_attn==2.8.3
-huggingface_hub==0.36.0
-imageio==2.36.0
-jaxtyping==0.3.4
-numpy
+# Core ML/DL frameworks
+accelerate==1.8.1
+diffusers==0.35.2
+peft==0.17.1
+megatron-core==0.12.1
+transformers==4.57.1
+ninja
+
+# Data processing
+numpy>=2.0
+pandas==2.2.3
+pillow==11.1.0
+opencv-python==4.11.0.86
+opencv-python-headless==4.11.0.86
+scikit-image==0.25.2
+scipy==1.15.3
+
+# Video/Media processing
+av==15.0.0
+decord==0.6.0
+imageio[pyav,ffmpeg]==2.37.0
+imageio-ffmpeg==0.6.0
+tifffile==2025.5.10
+ffmpegcv
+
+# Configuration/Hydra
+hydra-core==1.3.2
 omegaconf==2.3.0
-Pillow==12.0.0
-pymeshlab
-safetensors==0.7.0
-sageattention==1.0.6
-rembg
-tqdm==4.66.5
-transformers==4.37.2
-trimesh==4.4.7
-
-# --- Gradio & HF Spaces ---
-gradio
-spaces
-
-# --- External Packages (from git) ---
-# cubvh is installed dynamically in app.py to avoid build isolation issues.
-
-flash-attn @ https://huggingface.co/strangertoolshf/flash_attention_2_wheelhouse/resolve/main/wheelhouse-flash_attn-2.8.3/linux_x86_64/torch2.5/cu12/abiFALSE/cp312/flash_attn-2.8.3+cu12torch2.5cxx11abiFALSE-cp312-cp312-linux_x86_64.whl
+pyyaml==6.0.2
+
+# Tokenizers and text processing
+sentencepiece==0.2.0
+tiktoken==0.9.0
+tokenizers==0.22
+ftfy==6.3.1
+regex==2024.11.6
+
+# Data loading and storage
+webdataset==0.2.111
+zarr==2.18.3
+tensorstore==0.1.45
+numcodecs==0.13.1
+
+# AWS/S3 support
+boto3==1.38.31
+botocore==1.38.46
+s3transfer==0.13.1
+fsspec==2025.5.1
+
+# Model storage and optimization
+safetensors==0.5.3
+
+# Utilities
+einops==0.8.1
+attrs==25.3.0
+attr==0.3.2
+tyro==0.9.32
+tqdm==4.67.1
+psutil==7.0.0
+loguru==0.7.3
+rich==14.0.0
+matplotlib==3.10.7
+
+# Infrastructure
+fvcore==0.1.5.post20221221
+iopath==0.1.10
+huggingface-hub==0.35.3
+
+# Inference extras (moved from INSTALL.md step 7)
+pyquaternion
+lru-dict
+OpenEXR==3.2.3
+jaxtyping
+portalocker
+tabulate
+termcolor
+braceexpand
+fasteners
+lazy-loader>=0.4
+hatchling
+hatch-vcs>=0.4
+editables
+
+# Testing
+coverage==7.9.2
+pytest==8.4.1
+pytest-cov==6.2.1
+pytest-mock==3.14.1
+pytest-random-order==1.2.0
+pytest-xdist==3.8.0
+ipdb==0.13.13
+ipython==8.37.0
+
+# Standard dependencies (required by other packages)
+annotated-types==0.7.0
+asttokens==3.0.0
+certifi==2025.7.9
+charset-normalizer==3.4.2
+decorator==5.2.1
+executing==2.2.0
+filelock==3.28.0
+idna==3.10
+importlib-metadata==8.7.0
+jedi==0.19.2
+jinja2==3.1.6
+jmespath==1.0.1
+markupsafe==3.0.2
+ninja==1.11.1.4
+nvidia-ml-py==12.535.133
+orjson==3.11.3
+parso==0.8.4
+pexpect==4.9.0
+platformdirs==4.3.8
+prompt-toolkit==3.0.51
+protobuf==4.25.3
+ptyprocess==0.7.0
+pure-eval==0.2.3
+pydantic==2.11.7
+pydantic-core==2.33.2
+pygments==2.19.2
+pynvml==12.0.0
+pyparsing==3.2.3
+python-dateutil==2.9.0.post0
+pytz==2025.2
+requests==2.32.4
+setuptools==80.9.0
+six==1.17.0
+stack-data==0.6.3
+tomli==2.2.1
+tomlkit==0.13.3
+traitlets==5.14.3
+typing-extensions==4.14.1
+tzdata==2025.2
+urllib3==2.5.0
+wrapt==1.17.2
+zipp==3.23.0