test

requirements.txt

@@ -1,11 +1,47 @@
 --extra-index-url https://download.pytorch.org/whl/cu118
 
+
 torch==2.1.0
 torchaudio==2.1.0
 torchvision==0.16.0
 xformers==0.0.22.post7
-kolors@git+https://github.com/Kwai-Kolors/Kolors.git#egg=038818d
-https://huggingface.co/spaces/xinjjj/ImgRoboAssetGen/resolve/main/wheels/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl
-https://huggingface.co/spaces/xinjjj/ImgRoboAssetGen/resolve/main/wheels/kaolin-0.16.0-cp310-cp310-linux_x86_64.whl
-https://huggingface.co/spaces/xinjjj/ImgRoboAssetGen/resolve/main/wheels/nvdiffrast-0.3.3-cp310-cp310-linux_x86_64.whl
+dataclasses_json
+easydict
+opencv-python>4.5
+cpplint==2.0.0
+pre-commit==2.13.0
+imageio==2.36.1
+imageio-ffmpeg==0.5.1
+rembg==2.0.61
+trimesh==4.4.4
+moviepy==1.0.3
+pymeshfix==0.17.0
+igraph==0.11.8
+pyvista==0.36.1
+openai==1.58.1
+spconv-cu118==2.3.8
+transformers==4.42.4
+gradio_litmodel3d==0.0.1
+gradio==5.12.0
+sentencepiece==0.2.0
+diffusers==0.31.0
+xatlas==0.0.9
+onnxruntime==1.20.1
+tenacity==8.2.2
+pytorch-lightning==2.1.0
+gsplat==1.5.0
+accelerate==0.33.0
+basicsr==1.4.2
+realesrgan==0.3.0 
+flash-attn==2.6.3
+pydantic==2.9.2
 https://github.com/Dao-AILab/flash-attention/releases/download/v2.6.3/flash_attn-2.6.3+cu118torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
+utils3d@git+https://github.com/EasternJournalist/utils3d.git@9a4eb15e4021b67b12c460c7057d642626897ec8
+clip@git+https://github.com/openai/CLIP.git
+kolors@git+https://github.com/Kwai-Kolors/Kolors.git#egg=038818d
+segment-anything@git+https://github.com/facebookresearch/segment-anything.git#egg=dca509f
+
+wheels/kaolin-0.16.0-cp310-cp310-linux_x86_64.whl
+wheels/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl
+wheels/nvdiffrast-0.3.3-cp310-cp310-linux_x86_64.whl
+