# ============================================================
# ComfyUI HunyuanWorld - Requirements
# Converted from docker/HunyuanWorld.yaml (Linux conda env)
# Compatible with Windows + ComfyUI portable
# ============================================================
# NOTE: PyTorch, torchvision, torchaudio, CUDA libs are already
# included in ComfyUI portable. Do NOT reinstall them.
# ============================================================

# --- Core ML / Diffusion ---
diffusers==0.34.0
transformers==4.51.0
accelerate==1.6.0
safetensors==0.5.3
huggingface-hub==0.30.2
peft==0.15.0
tokenizers==0.21.1
sentencepiece==0.2.0

# --- Image / Vision ---
opencv-python==4.11.0.86
opencv-python-headless==4.11.0.86
Pillow>=11.0.0
albumentations==0.5.2
imageio==2.37.0
imageio-ffmpeg==0.4.9
scikit-image==0.24.0
kornia==0.8.0
kornia-rs==0.1.8
timm==1.0.13
open-clip-torch==2.30.0

# --- 3D / Geometry ---
open3d>=0.18.0
trimesh>=4.6.1
plyfile==1.1
py360convert==1.0.3
shapely==2.0.7

# --- Numeric / Scientific ---
numpy==1.24.1
scipy==1.15.2
scikit-learn==1.6.1
pandas==2.2.3
numba==0.61.2
llvmlite==0.44.0

# --- Deep Learning Utilities ---
einops==0.4.1
pytorch-lightning==2.4.0
torchmetrics==1.7.1
lightning-utilities==0.14.3

# --- Face / Super-Resolution (Real-ESRGAN) ---
facexlib==0.3.0
gfpgan==1.3.8

# --- Segmentation ---
segment-anything==1.0

# --- Object Detection ---
ultralytics==8.3.74
ultralytics-thop==2.0.14

# --- ONNX Runtime (for ZIM segmentation) ---
onnx==1.17.0
onnxruntime-gpu==1.21.1

# --- Config / CLI ---
omegaconf==2.1.2
hydra-core==1.1.0
antlr4-python3-runtime==4.8
easydict==1.9
addict==2.4.0
pyyaml>=6.0

# --- Data Loading ---
webdataset==0.2.100
braceexpand==0.1.7
lmdb==1.6.2
h5py==3.7.0
av==14.3.0

# --- Visualization / Logging ---
matplotlib==3.10.1
seaborn==0.13.2
tabulate==0.9.0
rich==14.0.0
loguru==0.7.3
icecream==2.1.2
coloredlogs==15.0.1
tqdm==4.67.1

# --- Text Processing ---
ftfy==6.1.1
regex==2022.6.2
inflect==5.6.0

# --- Networking / Async ---
aiohttp==3.11.16
aiohappyeyeballs==2.6.1
aiosignal==1.3.2
frozenlist==1.6.0
multidict==6.4.3
yarl==1.20.0
propcache==0.3.1
async-timeout==5.0.1

# --- General Utilities ---
filelock>=3.17.0
fsspec==2025.3.2
packaging==24.2
psutil==7.0.0
joblib==1.4.2
cloudpickle==3.1.1
six==1.17.0
future==1.0.0
colorama==0.4.6
platformdirs==4.3.7
importlib-metadata==8.6.1
zipp==3.21.0
attrs==25.3.0
certifi>=2025.1.31
charset-normalizer>=3.3.2
idna>=3.7
urllib3>=2.3.0
requests>=2.32.3
pysocks>=1.7.1
sympy>=1.13.1
networkx>=3.4.2
markupsafe>=3.0.2
jinja2>=3.1.6

# --- Misc ---
cython==3.0.11
cmake
protobuf==5.29.3
flatbuffers==25.2.10
grpcio==1.71.0
absl-py==2.2.2
yapf==0.43.0
filterpy==1.4.5
imgaug==0.4.0
werkzeug==3.1.3
diffdist==0.1
submitit==1.4.2
lazy-loader==0.4
tifffile==2025.3.30
contourpy==1.3.2
cycler==0.12.1
fonttools==4.57.0
kiwisolver==1.4.8
pyparsing==3.2.3
python-dateutil==2.9.0.post0
pytz==2025.2
tzdata==2025.2
tomli==2.2.1
markdown==3.8
markdown-it-py==3.0.0
mdurl==0.1.2
pygments==2.19.1
py-cpuinfo==9.0.0
humanfriendly==10.0
asttokens==3.0.0
executing==2.2.0
wcwidth==0.2.13
threadpoolctl==3.6.0
termcolor==3.0.1
wldhx-yadisk-direct==0.0.6
qwen-vl-utils==0.0.8
decord==0.6.0

# --- Git-based installs (run separately) ---
# pip install git+https://github.com/microsoft/MoGe.git
# pip install git+https://github.com/facebookresearch/pytorch3d.git

# ============================================================
# SKIPPED (Linux-only / already in ComfyUI / won't build on Windows):
# - flash-attn==2.7.4.post1      (Linux-only, requires CUDA build)
# - triton==3.2.0                (Linux-only)
# - natten==0.14.4               (Linux-only, needs custom build)
# - xformers==0.0.28.post2       (use ComfyUI's bundled version)
# - sageattention==1.0.6         (Linux-only)
# - nvidia-* packages            (bundled with PyTorch CUDA)
# - torch, torchvision, torchaudio (use ComfyUI's bundled version)
# - tensorboard / tb-nightly     (optional, not needed for inference)
# ============================================================
