Use complete requirements from official UltraPixel

requirements.txt

@@ -1,17 +1,26 @@
+--find-links https://download.pytorch.org/whl/torch_stable.html
 torch>=2.0.0
 torchvision>=0.15.0
-diffusers>=0.27.0
-transformers>=4.36.0
 accelerate>=0.25.0
+transformers>=4.30.0
+kornia>=0.7.0
+insightface>=0.7.3
+opencv-python>=4.8.1.78
+tqdm>=4.66.1
+matplotlib>=3.7.4
+webdataset>=0.2.79
+wandb>=0.16.2
+munch>=4.0.0
+onnxruntime>=1.16.3
+einops>=0.7.0
+onnx2torch>=1.5.13
+numpy>=1.23.5
+diffusers>=0.27.0
 safetensors>=0.4.0
 gradio>=4.19.0
 pillow>=10.0.0
-numpy>=1.24.0
 pyyaml>=6.0
 huggingface_hub>=0.20.0
-opencv-python>=4.8.0
-einops>=0.7.0
 omegaconf>=2.3.0
-matplotlib>=3.7.0
-git+https://github.com/ildoonet/pytorch-gradual-warmup-lr.git
-torchtools>=0.1.0
+warmup-scheduler @ git+https://github.com/ildoonet/pytorch-gradual-warmup-lr.git
+torchtools @ git+https://github.com/pabloppp/pytorch-tools